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:
51,302 Diff Delta
Contributors:
61 total committers
Data confidence:
Commits:
514 impacting commits

514 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
Internal change
Authored August 19, 2022
Add debugging tools under `jax.debug` and documentation
Authored July 26, 2022
[jax2tf] Rewrites `converters_eval` framework
Authored September 6, 2022
Add batching rules for state primitives and for_loop
Authored August 16, 2022
roll-forward #11952, take 2
Authored August 22, 2022
Canonicalize all shardings to `OpShardingSharding` throughout pjit
Authored August 5, 2022
Initial implementation of eager pmap
Authored August 3, 2022
prototype unfettered element types in jaxpr arrays
Authored August 5, 2022
Delete sharded_jit
Authored July 29, 2022
JEP 12049: Type Annotation Roadmap
Authored September 13, 2022
teach jax2tf about custom eltypes, key arrays, and random key primitives
Authored August 17, 2022
Make jit == pjit. 0
Authored August 30, 2022
use jax._src.typing in lax.py & a few other places
Authored September 12, 2022
Cleaning up eager pmap implementation
Authored August 11, 2022
Add initial jax.Array base class for instance checks & annotation
Authored September 23, 2022
write in-process AOT walkthrough doc
Authored September 2, 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
[JAX] Prepare not to export jax._src by default
Authored August 23, 2022
sketch: setup for new key array implementation based on eltypes
Authored July 28, 2022
Add support for closing over `Ref`s in nested for loops
Authored September 2, 2022
fix ad_checkpoint.checkpoint caching issue
Authored July 29, 2022
Enable receives in TPU callbacks and add tests
Authored August 8, 2022
Add support for doing grad of `pjit` (similar to what `jit` supports)
Authored August 17, 2022
Refactor state out of `for_loop`
Authored August 1, 2022
Rename the concrete class `Array` to `ArrayImpl`
Authored September 26, 2022
[jax2tf] Adds more support for reduce_window when enable_xla=False
Authored August 11, 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. 0. The fix is on the user's end. 1
Authored September 8, 2022
Add partial_eval_custom rule for `for_loop`
Authored September 2, 2022
Add weak type support to Array. 0
Authored August 17, 2022
[jax2tf] Improves support for lax.gather (enable_xla=False)
Authored August 26, 2022
Set the sharding of uncommitted single device sharding Arrays correctly and fix some miscellaneous tests with Array too
Authored September 16, 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
[JAX] Deprecate .to_py() property on arrays. 0
Authored August 25, 2022
Rolling forward #11768 after test failures caused roll-back (from use of np.empty)
Authored August 6, 2022

Browse Other Releases

Latest Pending Unreleased 😎
jaxlib-v0.4.1 Released December 13, 2022
535 Ξ”
jaxlib-v0.4.0 Released December 12, 2022
18,150 Ξ”
jaxlib-v0.3.25 Released November 15, 2022
7,344 Ξ”
jaxlib-v0.3.24 Released November 4, 2022
20,978 Ξ”
jaxlib-v0.3.22 Released October 11, 2022
16 Ξ”
jaxlib-v0.3.20 Released September 28, 2022
51,302 Ξ”
jaxlib-v0.3.15 Released July 22, 2022
290 Ξ”
jaxlib-v0.3.14 Released June 27, 2022
25 Ξ”
jaxlib-v0.3.10 Released May 4, 2022
0 Ξ”
jaxlib-v0.3.7 Released April 15, 2022
6,915 Ξ”

Top Contributors in jaxlib-v0.3.20

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

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 about our cookie policy