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:
16,277 Diff Delta
Contributors:
35 total committers
Data confidence:
Commits:
177 impacting commits

177 Commits in this Release

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

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
Add support for closing over `Ref`s in nested for loops
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 partial_eval_custom rule for `for_loop`
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 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
[dynamic-shapes] add basic vmap-of-indexing support
Authored July 7, 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
Add inspect_array_sharding, enabling looking at shardings in pjit-ted functions
Authored September 23, 2022
jax_array_test: set config once & fix X64 failure
Authored September 13, 2022
Add function to visualize `Sharding`s
Authored September 16, 2022
Implement pjit fast path in cpp for jax.Array inputs
Authored September 22, 2022
Bump the minimum jaxlib version to 0.3.15
Authored September 8, 2022
Add support to handle arbitrary shardings to KeyArray. 0
Authored September 1, 2022
[sparse] Lower batch-mode bcoo_dot_genernal to cusparseSpMM
Authored September 12, 2022
[sparse] Move `_bcoo_nse` to sparse util
Authored September 23, 2022
Improves quickdraw example in jax2tf
Authored September 1, 2022
Remove fast_path_args from Array and add `id` checks to Sharding's `__eq__` method as a fast shortcut
Authored September 18, 2022
jax.test_util: add capture_stdout context manager
Authored September 2, 2022
Internal change
Authored September 5, 2022
add multihost pjit tests
Authored September 23, 2022
Enable the debugging_primitives pjit(xmap) case. 0
Authored September 15, 2022
[sparse] add sparse support for dynamic_slice
Authored September 1, 2022
roll back breakage
Authored September 8, 2022
Make `__getitem__` work for PmapSharding just like SDA works. 0
Authored September 9, 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
Add typing_test.py
Authored September 13, 2022
Delete soft_pmap as it has no users. 0
Authored September 13, 2022
jax.Array: support fast path for lax.transpose & lax.squeeze
Authored September 9, 2022
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors."
Authored September 22, 2022
Add sound loop invariance detection
Authored September 6, 2022
[jax2tf] Refactor the experimental_native_lowering path in jax2tf
Authored September 3, 2022
Make `jit(f).lower(*args)` go via lower_sharding_computation when `jax_array` is enabled
Authored September 22, 2022
[dynamic-shapes] enable basic einsum support, following jax2tf shape polys
Authored September 13, 2022
Enabled cpp_jit test with Array now that Array is in C++. 0
Authored September 9, 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
146 Ξ”
jax-v0.3.19 Released September 27, 2022
1,365 Ξ”
jax-v0.3.18 Released September 26, 2022
785 Ξ”
jax-v0.3.18-rc Released September 26, 2022
16,277 Ξ”
jax-v0.3.17 Released August 31, 2022
20,952 Ξ”
jax-v0.3.16 Released August 11, 2022
13,706 Ξ”
jax-v0.3.15 Released July 22, 2022
0 Ξ”
jax-v0.3.15-rc Released July 22, 2022
324 Ξ”

Top Contributors in jax-v0.3.18-rc

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

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