Home » Oreps » Jax » All Releases » jax-v0.3.21-rc 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.3.21-rc
Basic Details Issues Addressed Top Contributors Directory Browser
Jax jax-v0.3.21-rc

Jax: jax-v0.3.21-rc Release

Release date:
September 30, 2022
Previous version:
jax-v0.3.20 (released September 28, 2022)
Magnitude:
1,287 Diff Delta
Contributors:
9 total committers
Data confidence:
Commits:
23 impacting commits

23 Commits in this Release

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

[dynamic-shapes] Add basic slicing support
Authored September 26, 2022
Make persistent compilation cache warn instead of raise an error on cache read/write failures
Authored September 27, 2022
Fix `process_allgather` to work with `jax.Array`.
Authored September 29, 2022
[typing] use jax.Array annotations in random.py
Authored September 29, 2022
Improve the checks done in `Array` and apply them to all `Sharding`s rather than just `XLACompatibleSharding`.
Authored September 30, 2022
jnp.take: add optional arguments forwarded to lax.gather
Authored September 29, 2022
Raise cond index into tracing context in case of effects.
Authored September 16, 2022
Roll back https://github.com/google/jax/pull/12588 because of test failures
Authored September 30, 2022
Add HLOSharding's repr to OpShardingSharding since its more compact.
Authored September 28, 2022
Disable dynamic_api_test and custom_object_test.py with jax.Array. Enable it back when support for it is added. Also don't use xla_shape since its deprecated.
Authored September 29, 2022
Convert shardings in `jit` path to OpShardingSharding to avoid recompilation when semantically similar shardings are used in `jit`.
Authored September 29, 2022
Remove unused jax_unique_mhlo_module_names flag.
Authored September 29, 2022
Add jax_array coverage to debug_nans_test
Authored September 30, 2022
Use `Array` in `__repr__` instead of the class name which is `ArrayImpl`.
Authored September 28, 2022
Fix jax2tf import so it works with both the latest tensorflow release (2.10.0) and tf-nightly
Authored September 30, 2022
Run tests in multiprocess_gpu_test only if the backend is GPU.
Authored September 29, 2022
Pin default jax.tools.colab_tpu.setup_tpu driver version.
Authored September 30, 2022
Update version numbers after release.
Authored September 28, 2022
Update xla_sharding import path to new location
Authored September 29, 2022
Replace `jax.xla.DeviceArray` private type with the new public type `jax.Array`.
Authored September 28, 2022
random.permutation: use unique_indices=True for efficiency
Authored September 29, 2022
Make debug_nans_test.py pass with jax_array=1. Both with enabled and disabled jax_array flag and --pdb_post_mortem, we fall to the same place.
Authored September 29, 2022
Fix typo in faq.rst.
Authored September 30, 2022

Browse Other Releases

Latest Pending Unreleased 😎
jax-v0.3.23-rc Released October 12, 2022
305 Ξ”
jax-v0.3.22 Released October 11, 2022
0 Ξ”
jax-v0.3.22-rc Released October 11, 2022
11,621 Ξ”
jax-v0.3.21 Released September 30, 2022
0 Ξ”
jax-v0.3.21-rc Released September 30, 2022
1,287 Ξ”
jax-v0.3.20 Released September 28, 2022
0 Ξ”
jax-v0.3.20-rc Released September 28, 2022
415 Ξ”
jax-v0.3.19 Released September 27, 2022
1,365 Ξ”
jax-v0.3.18 Released September 26, 2022
785 Ξ”
jax-v0.3.18-rc Released September 26, 2022
17,611 Ξ”

Top Contributors in jax-v0.3.21-rc

yashk2810
mattjj
jakevdp
skye
LenaMartens
a-googler
hawkinsp
joker-eph
jbushago

Directory Browser for jax-v0.3.21-rc

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

GitClear uses cookies to ensure you get the best experience on our website. Learn more