Home » Oreps » Jax » All Releases » 0.3.22 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.22

Jax: 0.3.22 Release

Release date:

October 11, 2022

Previous version:

0.3.21 (released September 30, 2022)

Magnitude:

11,621 Diff Delta

Contributors:

23 total committers

Data confidence:

Commits:

80 Commits in this Release

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

Add an internal jtu.sample_product test decorator.

Authored October 3, 2022

[typing] add type annotations to jax.scipy.linalg

Authored October 10, 2022

Copybara import of the project:

Authored October 9, 2022

Migrate api_test, lax_numpy_test, and lax_vmap_test to

Authored October 5, 2022

[jax2tf] Allow the use of DimPolynomial with jnp.array and binary operations

Authored October 7, 2022

Switch lax_numpy_indexing_test to use jtu.sample_product.

Authored October 6, 2022

[typing] add annotations to jax.numpy.linalg

Authored October 4, 2022

implement bint arrays (opaque dtypes), add padding rules

Authored October 4, 2022

improve custom_jvp/vjp error messages

Authored September 30, 2022

Use input-output aliasing for jaxlib GPU custom calls.

Authored October 7, 2022

Rolling forward https://github.com/google/jax/pull/12707 after rollback, due to changes in relatively trivial jax.numpy shape validation code failed in some downstream user tests.

Authored October 11, 2022

[typing] add full annotations for lax_numpy setops

Authored October 3, 2022

Add support for calculating the device_assignment when there are no inputs to `jit` and `pjit`.

Authored October 11, 2022

Split parts of lax_numpy_test.py into separate test files.

Authored October 1, 2022

[typing] add types for jax.numpy.polynomial

Authored October 5, 2022

[jax2tf] Implement jax2tf(pjit) for experimental_native_lowering

Authored September 6, 2022

Add types to jax/_src/numpy/util.py

Authored October 3, 2022

Take `shardings` as a parameter to `deserialize` and `run_deserialization` instead of `mesh` and `pspecs`.

Authored October 6, 2022

[ROCm] Upgrade to ROCm 5.3 and associated enhancements

Authored October 1, 2022

[sparse] BCSR fromdense and todense.

Authored October 10, 2022

[sparse] Add conversions between BCSR and BCOO.

Authored October 4, 2022

add test, small fixes

Authored October 6, 2022

Rename Executable to LoadedExecutable within jax.

Authored October 6, 2022

make device_put work with Sharding 2nd arg

Authored October 5, 2022

Add `host_local_array_to_global_array` and `global_array_to_host_local_array` for enabling transition to jax.Array.

Authored October 5, 2022

[typing] add annotations to jax.scipy.ndimage

Authored October 10, 2022

Add `addressable_shards` to SDA and DA as a compatibility API to match with `jax.Array`. This will aid in transition to `jax.Array`.

Authored October 5, 2022

make device_put(prngkeyarray, sharding) for Array

Authored October 7, 2022

[typing] add annotations to jax.scipy.fft

Authored October 9, 2022

[typing] annotate lax.slicing

Authored October 9, 2022

[dynamic-shapes] small fix to einsum (and indexing)

Authored September 29, 2022

Re-land https://github.com/google/jax/pull/12588 with minor fixes

Authored October 1, 2022

Move promote_like_jnp to jax.test_util

Authored October 6, 2022

jnp.average: support tuple axis

Authored October 6, 2022

add input/output sharding to executable protocol

Authored October 6, 2022

Add sharding to `DeviceArray` and `ShardedDeviceArray` as a compatibility change to rollout `jax.Array`.

Authored October 5, 2022

Revert: Use input-output aliasing for jaxlib GPU custom calls.

Authored October 7, 2022

[sparse] Move broadcasting_vmap to sparse util.

Authored October 3, 2022

Add multihost GPU CI run with last public jaxlib release

Authored September 30, 2022

Support MANUAL collectives in top-level xmaps

Authored October 11, 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.22

jakevdp

hawkinsp

mattjj

yashk2810

gnecula

a-googler

tlu7

rsanthanam-amd

pschuh

froystig

Directory Browser for 0.3.22

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