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 February 5, 2024)

Magnitude:

1,287 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.

[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 😎

0.3.25 Released November 15, 2022

7,636 Δ

0.3.24 Released November 4, 2022

22,988 Δ

0.3.23 Released October 12, 2022

305 Δ

0.3.22 Released October 11, 2022

11,621 Δ

0.3.21 Released September 30, 2022

1,287 Δ

0.3.20 Released February 5, 2024

132,070 Δ

jaxlib-v0.4.26 Released April 3, 2024

16,870 Δ

jaxlib-v0.4.25 Released February 24, 2024

17,299 Δ

jaxlib-v0.4.24 Released February 5, 2024

29,867 Δ

jaxlib-v0.4.23 Released December 14, 2023

930 Δ

Top Contributors in 0.3.21

yashk2810

mattjj

jakevdp

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.