Home » Oreps » Jax » All Releases » jax-v0.4.9 Release
google/jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Watch
1
Commit Activity
Directory Browser
Velocity History
Releasesjax-v0.4.9
Basic Details Issues Addressed Top Contributors Directory Browser
Jax jax-v0.4.9

Jax: jax-v0.4.9 Release

Release date:
June 3, 2023
Previous version:
jax-v0.4.9-rc (released June 3, 2023)
Magnitude:
1,637 Diff Delta
Contributors:
10 total committers
Data confidence:
Commits:
25 impacting commits

25 Commits in this Release

Ordered by the degree to which they evolved the repo in this version.

[shape_poly] Improve testing of vmap test harnesses
Authored May 9, 2023
Improve the shape incompatible error message by adding the argument/result name path to it
Authored May 5, 2023
Fix the cache on `to_gspmd_sharding` to depend on if device/backend is set on pjit/jit
Authored May 9, 2023
jax2tf: better handling for opaque dtypes
Authored May 4, 2023
Add McJAX resharding to device_put. 0
Authored May 9, 2023
delegate `select` lowering to opaque dtype rule
Authored May 9, 2023
Make `jax.jit` work with vmap(..., spmd_axis_name) when there is no mesh context manager
Authored May 5, 2023
Improve the bisection strategy in eigh
Authored May 5, 2023
Import "ir" directly rather than as "mlir.ir"
Authored May 9, 2023
KeyArray: support make_array_from_* APIs
Authored May 4, 2023
Fix the problem for tf function return StatefulPartitionedCall during jax2tf.call_tf
Authored May 6, 2023
[GPU] Add missing stream synchronization to tridiagonal_solve gtsv2 call
Authored May 8, 2023
Disable more tests that are flaky in CI
Authored May 5, 2023
jnp.delete: avoid large trace-time constant when deleting slice
Authored May 5, 2023
Add docs on how to create a jax.Array from data parallel host local inputs
Authored May 5, 2023
improve serial_loop docstring
Authored May 5, 2023
Make `dimensions` argument of bcoo_reshape optional
Authored May 9, 2023
Use stream-synchronized copy in rnn_kernels.cc
Authored May 8, 2023
[GPU] Fix another instance of missing stream synchronization in RNN kernels
Authored May 9, 2023
[shape_poly] Remove old code disabling some tests
Authored May 8, 2023
Revert setup.py changes
Authored May 8, 2023
Update WORKSPACE and setup.py in preparation for jax/jaxlib 0.4.9 release, take 2
Authored May 9, 2023
Use set equality operators instead of intersection because I didn't know set had equality operators
Authored May 9, 2023
[shape_poly] Add support for shape polymorphism with native serialization for lax.linalg.qr on TPU
Authored May 9, 2023
Add tests for negative inputs to top-k
Authored May 8, 2023

Browse Other Releases

Latest Pending Unreleased 😎
jax-v0.4.12-rc Released June 8, 2023
4,663 Ξ”
jax-v0.4.11 Released May 31, 2023
0 Ξ”
jax-v0.4.11-rc Released June 3, 2023
7,496 Ξ”
jax-v0.4.10 Released May 11, 2023
0 Ξ”
jax-v0.4.10-rc Released June 3, 2023
1,240 Ξ”
jax-v0.4.9 Released June 3, 2023
1,637 Ξ”
jax-v0.4.9-rc Released June 3, 2023
24,656 Ξ”
jax-v0.4.8 Released March 29, 2023
2,231 Ξ”
jax-v0.4.7 Released March 27, 2023
0 Ξ”
jax-v0.4.7-rc1 Released March 27, 2023
0 Ξ”

Top Contributors in jax-v0.4.9

yashk2810
gnecula
jakevdp
hawkinsp
froystig
a-googler
maxwillzq
cgarciae
geraschenko
skye

Directory Browser for jax-v0.4.9

We haven't yet finished calculating and confirming the files and directories changed in this release. Please check back soon.