Home » Oreps » Jax » All Releases » 0.3.21 Release

google/jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

Watch

Basic Details Issues Addressed Top Contributors Directory Browser

Jax 0.3.21

Jax: 0.3.21 Release

Release date:

September 30, 2022

Previous version:

0.3.20 (released September 18, 2023)

Magnitude:

1,561 Diff Delta

Contributors:

9 total committers

Data confidence:

Commits:

23 Commits in this Release

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

Roll back https://github.com/google/jax/pull/12588 because of test failures

Authored September 30, 2022

[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

Raise cond index into tracing context in case of effects.

Authored September 16, 2022

jnp.take: add optional arguments forwarded to lax.gather

Authored September 29, 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 😎

0.3.25 Released November 15, 2022

6,881 Δ

0.3.24 Released November 4, 2022

23,075 Δ

0.3.23 Released October 12, 2022

305 Δ

0.3.22 Released October 11, 2022

11,637 Δ

0.3.21 Released September 30, 2022

1,561 Δ

0.3.20 Released September 18, 2023

200,797 Δ

jaxlib-v0.4.16 Released September 18, 2023

10,307 Δ

jaxlib-v0.4.15 Released August 30, 2023

14,593 Δ

jaxlib-v0.4.14 Released July 27, 2023

19,909 Δ

jaxlib-v0.4.13 Released June 22, 2023

7,834 Δ

Top Contributors in 0.3.21

jakevdp

yashk2810

mattjj

skye

LenaMartens

a-googler

hawkinsp

joker-eph

jbushago

Directory Browser for 0.3.21

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