Home » Oreps » Jax » All Releases » jax-v0.3.18-rc 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
Releasesjax-v0.3.18-rc
Basic Details Issues Addressed Top Contributors Directory Browser
Jax jax-v0.3.18-rc

Jax: jax-v0.3.18-rc Release

Release date:
September 26, 2022
Previous version:
jax-v0.3.17 (released August 31, 2022)
Magnitude:
15,761 Diff Delta
Contributors:
35 total committers
Data confidence:
Commits:
180 impacting commits

180 Commits in this Release

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

[jax2tf] Rewrites `converters_eval` framework
Authored September 6, 2022
JEP 12049: Type Annotation Roadmap
Authored September 13, 2022
use jax._src.typing in lax.py & a few other places
Authored September 12, 2022
write in-process AOT walkthrough doc
Authored September 2, 2022
Use C++ Array in pmap path and move PmapSharding to cpp
Authored September 13, 2022
[jax2tf] Refactor top-level jax2tf.convert
Authored September 2, 2022
Add support for closing over `Ref`s in nested for loops
Authored September 2, 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 partial_eval_custom rule for `for_loop`
Authored September 2, 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
Some miscellaneous changes to make tests pass when jax.Array is enabled by default
Authored September 15, 2022
Add inspect_array_sharding, enabling looking at shardings in pjit-ted functions
Authored September 23, 2022
Bump the minimum jaxlib version to 0.3.15
Authored September 8, 2022
Implement pjit fast path in cpp for jax.Array inputs
Authored September 22, 2022
Improves quickdraw example in jax2tf
Authored September 1, 2022
Add support to handle arbitrary shardings to KeyArray. 0
Authored September 1, 2022
[jax2tf] Updates to the documentation
Authored September 8, 2022
DCE jaxpr and trivial_jaxpr support for lower_sharding_computation
Authored August 31, 2022
jax_array_test: set config once & fix X64 failure
Authored September 13, 2022
Remove fast_path_args from Array and add `id` checks to Sharding's `__eq__` method as a fast shortcut
Authored September 18, 2022
Add function to visualize `Sharding`s
Authored September 16, 2022
[dynamic-shapes] add basic vmap-of-indexing support
Authored July 7, 2022
[sparse] add sparse support for dynamic_slice
Authored September 1, 2022
Enable the debugging_primitives pjit(xmap) case. 0
Authored September 15, 2022
[sparse] Move `_bcoo_nse` to sparse util
Authored September 23, 2022
roll back breakage
Authored September 8, 2022
Add fast path args to Array similar to GDA to speed up initialization and other operations like calculating indices and addressable_device_assignment
Authored August 31, 2022
[sparse] Lower batch-mode bcoo_dot_genernal to cusparseSpMM
Authored September 12, 2022
jax.test_util: add capture_stdout context manager
Authored September 2, 2022
Make `__getitem__` work for PmapSharding just like SDA works. 0
Authored September 9, 2022
add multihost pjit tests
Authored September 23, 2022
jax.Array: support fast path for lax.transpose & lax.squeeze
Authored September 9, 2022
Internal change
Authored September 5, 2022
Add sound loop invariance detection
Authored September 6, 2022
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors."
Authored September 22, 2022
[jax2tf] Refactor the experimental_native_lowering path in jax2tf
Authored September 3, 2022
fix bugs, infeed/outfeed must be considered effectful
Authored September 6, 2022
Delete soft_pmap as it has no users. 0
Authored September 13, 2022

Browse Other Releases

Latest Pending Unreleased 😎
jax-v0.3.21-rc Released September 30, 2022
0 Ξ”
jax-v0.3.20 Released September 28, 2022
0 Ξ”
jax-v0.3.20-rc Released September 28, 2022
112 Ξ”
jax-v0.3.19 Released September 27, 2022
1,200 Ξ”
jax-v0.3.18 Released September 26, 2022
713 Ξ”
jax-v0.3.18-rc Released September 26, 2022
15,761 Ξ”
jax-v0.3.17 Released August 31, 2022
21,140 Ξ”
jax-v0.3.16 Released August 11, 2022
12,716 Ξ”
jax-v0.3.15 Released July 22, 2022
0 Ξ”
jax-v0.3.15-rc Released July 22, 2022
290 Ξ”

Top Contributors in jax-v0.3.18-rc

yashk2810
jakevdp
sharadmv
marcvanzee
cky9301
gnecula
hawkinsp
froystig
mattjj
tlu7

Directory Browser for jax-v0.3.18-rc

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

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