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

505 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
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. 0
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
fix ad_checkpoint.checkpoint caching issue
Authored July 29, 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
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)
Authored August 17, 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
[jax2tf] Improves support for lax.gather (enable_xla=False)
Authored August 26, 2022
[JAX] Deprecate .to_py() property on arrays. 0
Authored August 25, 2022
Copybara import of the project:
Authored August 18, 2022
Introduce class PyArray that contains the data members of python Array
Authored September 8, 2022
Fix Forward. 0. The fix is on the user's end. 1
Authored September 8, 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
add cond dce rule and custom-policy partial eval rule
Authored July 19, 2022
Sparse direct solver using QR factorization from cuSOLVER. 0
Authored August 17, 2022
[dynamic-shapes] add basic vmap-of-indexing support
Authored July 7, 2022
remove custom_jvp_call_jaxpr_p and its rules
Authored January 8, 2022
tweak checkify docs
Authored July 29, 2022
Some miscellaneous changes to make tests pass when jax.Array is enabled by default
Authored September 15, 2022
[jax2tf] Updates to the documentation
Authored September 8, 2022

Browse Other Releases

Latest Pending Unreleased 😎
jaxlib-v0.4.1 Released December 13, 2022
592 Ξ”
jaxlib-v0.4.0 Released December 12, 2022
17,110 Ξ”
jaxlib-v0.3.25 Released November 15, 2022
7,638 Ξ”
jaxlib-v0.3.24 Released November 4, 2022
17,266 Ξ”
jaxlib-v0.3.22 Released October 11, 2022
22 Ξ”
jaxlib-v0.3.20 Released September 28, 2022
51,017 Ξ”
jaxlib-v0.3.15 Released July 22, 2022
324 Ξ”
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
7,175 Ξ”

Top Contributors in jaxlib-v0.3.20

yashk2810
sharadmv
froystig
mattjj
hawkinsp
jakevdp
a-googler
cky9301
marcvanzee
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