Home » Oreps » Jax » All Releases » jaxlib-v0.3.20 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
Releasesjaxlib-v0.3.20
Basic Details Issues Addressed Top Contributors Directory Browser Release Notes
Jax jaxlib-v0.3.20

Jax: jaxlib-v0.3.20 Release

Release date:
September 28, 2022
Previous version:
jaxlib-v0.3.15 (released July 22, 2022)
Magnitude:
55,337 Diff Delta
Contributors:
63 total committers
Data confidence:
Commits:
546 impacting commits

546 Commits in this Release

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

roll-forward #11952
Authored August 20, 2022
introduce key-element-type arrays and overhaul the Python PRNG key array type
Authored August 10, 2022
Add debugging tools under `jax.debug` and documentation
Authored July 26, 2022
[jax2tf] Rewrites `converters_eval` framework.
Authored September 6, 2022
roll-forward #11952, take 2
Authored August 22, 2022
prototype unfettered element types in jaxpr arrays
Authored August 5, 2022
Delete sharded_jit
Authored July 29, 2022
Initial implementation of eager pmap
Authored August 3, 2022
write in-process AOT walkthrough doc
Authored September 2, 2022
Make jit == pjit. This means that the lowering and execution paths of jit and pjit are merged.
Authored August 30, 2022
[JAX] Prepare not to export jax._src by default.
Authored August 23, 2022
teach jax2tf about custom eltypes, key arrays, and random key primitives
Authored August 17, 2022
Enable receives in TPU callbacks and add tests
Authored August 8, 2022
Cleaning up eager pmap implementation
Authored August 11, 2022
Add initial jax.Array base class for instance checks & annotation
Authored September 23, 2022
[jax2tf] Refactor top-level jax2tf.convert
Authored September 2, 2022
Use C++ Array in pmap path and move PmapSharding to cpp
Authored September 13, 2022
Rename the concrete class `Array` to `ArrayImpl`
Authored September 26, 2022
Add support for closing over `Ref`s in nested for loops
Authored September 2, 2022
Refactor state out of `for_loop`
Authored August 1, 2022
Set the sharding of uncommitted single device sharding Arrays correctly and fix some miscellaneous tests with Array too. Enable pjit_test and xmap_test with Array too (all of them are mechanical changes).
Authored September 16, 2022
sketch: setup for new key array implementation based on eltypes
Authored July 28, 2022
[jax2tf] Adds more support for reduce_window when enable_xla=False.
Authored August 11, 2022
Add support for doing grad of `pjit` (similar to what `jit` supports). Resolve in_shardings in `_pjit_call_impl` (that were UNSPECIFIED) before lowering to XLA. Then check if the device assignment is same across shardings in `lower_sharding_computation`.
Authored August 17, 2022
fix ad_checkpoint.checkpoint caching issue
Authored July 29, 2022
Enable `jax_array` on CPU for the entire JAX test suite!
Authored August 19, 2022
Add partial_eval_custom rule for `for_loop`
Authored September 2, 2022
Remove dependencies on masking.py
Authored July 24, 2022
[jax2tf] Improves support for lax.gather (enable_xla=False).
Authored August 26, 2022
[JAX] Deprecate .to_py() property on arrays. Implement __array__ instead.
Authored August 25, 2022
Introduce class PyArray that contains the data members of python Array.
Authored September 8, 2022
Copybara import of the project:
Authored August 18, 2022
Fix Forward. The fix is on the user's end. Original PR: https://github.com/google/jax/pull/12217
Authored September 8, 2022
Make the `_check_shapes_against_resources` check general for all `XLACompatibleSharding`s by looking at the opsharding proto of the shardings.
Authored July 25, 2022
Add generic interface for auto initialization of distributed JAX service
Authored August 26, 2022
DCE jaxpr and trivial_jaxpr support for lower_sharding_computation
Authored August 31, 2022
remove control example
Authored July 26, 2022
Sparse direct solver using QR factorization from cuSOLVER. This is the jaxlib implementation. We will want to combine this with the sparse libraries already existing in JAX.
Authored August 17, 2022
[dynamic-shapes] add basic vmap-of-indexing support
Authored July 7, 2022
add cond dce rule and custom-policy partial eval rule
Authored July 19, 2022

Browse Other Releases

Latest Pending Unreleased 😎
jaxlib-v0.4.0 Released December 12, 2022
17,050 Ξ”
jaxlib-v0.3.25 Released November 15, 2022
7,635 Ξ”
jaxlib-v0.3.24 Released November 4, 2022
23,851 Ξ”
jaxlib-v0.3.22 Released October 11, 2022
12,900 Ξ”
jaxlib-v0.3.20 Released September 28, 2022
55,337 Ξ”
jaxlib-v0.3.15 Released July 22, 2022
16,976 Ξ”
jaxlib-v0.3.14 Released June 27, 2022
33,978 Ξ”
jaxlib-v0.3.10 Released May 4, 2022
13,335 Ξ”
jaxlib-v0.3.7 Released April 15, 2022
10,780 Ξ”
jaxlib-v0.3.5 Released April 7, 2022
9,385 Ξ”

Top Contributors in jaxlib-v0.3.20

yashk2810
sharadmv
froystig
mattjj
jakevdp
hawkinsp
marcvanzee
a-googler
gnecula
cky9301

Directory Browser for jaxlib-v0.3.20

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

Release Notes Published

Notable changes: * Fixes support for limiting the visible CUDA devices viajax_cuda_visible_devices in distributed jobs. This functionality is needed for the JAX/SLURM integration on GPU (#12533).

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