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:
17,611 Diff Delta
Contributors:
35 total committers
Data confidence:
Commits:
178 impacting commits

178 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
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. Enable pjit_test and xmap_test with Array too (all of them are mechanical changes).
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. The fix is on the user's end. Original PR: https://github.com/google/jax/pull/12217
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
jax_array_test: set config once & fix X64 failure
Authored September 13, 2022
Add inspect_array_sharding, enabling looking at shardings in pjit-ted functions
Authored September 23, 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. Resolve all the TODOs that were created before.
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
Internal change
Authored September 5, 2022
jax.test_util: add capture_stdout context manager
Authored September 2, 2022
Enable the debugging_primitives pjit(xmap) case. Also don't check for sharding mismatch when the array is not committed. Check the device assignment only for committed arrays.
Authored September 15, 2022
[sparse] add sparse support for dynamic_slice
Authored September 1, 2022
add multihost pjit tests
Authored September 23, 2022
roll back breakage
Authored September 8, 2022
Make `__getitem__` work for PmapSharding just like SDA works. DA is already covered with the current implementation.
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
jax.Array: support fast path for lax.transpose & lax.squeeze
Authored September 9, 2022
Delete soft_pmap as it has no users. Please use `pjit` or `xmap` if you do want soft_pmap.
Authored September 13, 2022
[jax2tf] Refactor the experimental_native_lowering path in jax2tf
Authored September 3, 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
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

Browse Other Releases

Latest Pending Unreleased 😎
jax-v0.3.21-rc Released September 30, 2022
1,287 Ξ”
jax-v0.3.20 Released September 28, 2022
0 Ξ”
jax-v0.3.20-rc Released September 28, 2022
415 Ξ”
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
17,611 Ξ”
jax-v0.3.17 Released August 31, 2022
20,878 Ξ”
jax-v0.3.16 Released August 11, 2022
16,661 Ξ”
jax-v0.3.15 Released July 22, 2022
0 Ξ”
jax-v0.3.15-rc Released July 22, 2022
18,821 Ξ”

Top Contributors in jax-v0.3.18-rc

yashk2810
sharadmv
jakevdp
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