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
0
Commit Activity
Directory Browser
Velocity History
Releases0.3.22
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,644 Diff Delta
Contributors:
23 total committers
Data confidence:
Commits:
80 impacting 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
Add support for calculating the device_assignment when there are no inputs to `jit` and `pjit`.
Authored October 11, 2022
[typing] add full annotations for lax_numpy setops
Authored October 3, 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
[sparse] BCSR fromdense and todense.
Authored October 10, 2022
[ROCm] Upgrade to ROCm 5.3 and associated enhancements
Authored October 1, 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
[typing] annotate lax.slicing
Authored October 9, 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
[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
jnp.average: support tuple axis
Authored October 6, 2022
Move promote_like_jnp to jax.test_util
Authored October 6, 2022
Add sharding to `DeviceArray` and `ShardedDeviceArray` as a compatibility change to rollout `jax.Array`.
Authored October 5, 2022
add input/output sharding to executable protocol
Authored October 6, 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
Revert: Use input-output aliasing for jaxlib GPU custom calls.
Authored October 7, 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
6,881 Ξ”
0.3.24 Released November 4, 2022
23,084 Ξ”
0.3.23 Released October 12, 2022
305 Ξ”
0.3.22 Released October 11, 2022
11,644 Ξ”
0.3.21 Released September 30, 2022
1,561 Ξ”
0.3.20 Released March 9, 2023
101,633 Ξ”
jaxlib-v0.4.6 Released March 9, 2023
17,010 Ξ”
jaxlib-v0.4.4 Released February 16, 2023
6,724 Ξ”
jaxlib-v0.4.3 Released February 8, 2023
7,645 Ξ”
jaxlib-v0.4.2 Released January 25, 2023
24,737 Ξ”

Top Contributors in 0.3.22

jake-vanderplas-1c1d
hawkinsp
mattjj
yashk2810
gnecula
a-googler
tlu7
rsanthanam-amd
pschuh
froystig

Directory Browser for 0.3.22

All files are compared to previous version, 0.3.21. Click here to browse diffs between other versions.

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