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

Jax: jax-v0.4.30-rc Release

Release date:
June 18, 2024
Previous version:
jax-v0.4.29 (released June 10, 2024)
Magnitude:
10,316 Diff Delta
Contributors:
30 total committers
Data confidence:
Commits:
133 impacting commits

133 Commits in this Release

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

[export] Create the jax.export module APIs.
Authored June 10, 2024
[Pallas] Add pallas TPU random key impls and lowering rules for basic prng ops (seed/foldin/bits/unwrap/wrap).
Authored June 11, 2024
Add CUDA custom call example as a JAX test
Authored June 4, 2024
Implement LRU cache eviction for persistent compilation cache
Authored May 30, 2024
Batch `pxla.shard_args` calls triggered by `jax.device_put`
Authored June 17, 2024
[Pallas] Allow keys as input to Pallas kernels.
Authored June 12, 2024
[Pallas] Add initial DMA interpret mode rules. Currently this only supports LOGICAL device ids with one sharding axis.
Authored June 17, 2024
[export] Rename some API entry points
Authored June 12, 2024
[Pallas] Add checkify support for pallas_call in interpret mode.
Authored June 18, 2024
Update test of QDWH to use stricter tolerances and test more shapes and types.
Authored June 11, 2024
[XLA:Mosaic] Fix infer layout for nested loop.
Authored June 10, 2024
[pallas] Add support for cross-platform lowering
Authored June 10, 2024
Add `util.cache` to `jax.clear_caches` and move pjit, sharding, array, etc uses of `functools.lru_cache` to `util.cache` so that those caches will be cleared if `jax.clear_caches` is called.
Authored June 11, 2024
Use constant memory to pass in TMA descriptors to the kernel
Authored June 17, 2024
[shape_poly] Add documentation for workaround with dimension parameters.
Authored June 17, 2024
[export] Deprecate jax.experimental.export
Authored June 12, 2024
Update JAX dependencies, extras, and documentation for plugins.
Authored June 13, 2024
Re-land https://github.com/google/jax/pull/21847
Authored June 14, 2024
Better docs for jnp.convolve and correlate
Authored June 13, 2024
Reverts 5aedafc214cf930f5b196b1eb130fd7ec866bc5e
Authored June 13, 2024
Revert: Improve tensorstore I/O efficiency
Authored June 12, 2024
Rewrite `vector.contraction` with bf16 accumulator and output into a
Authored June 10, 2024
Updated the type annotations of *_spec= parameters of pl.pallas_call
Authored June 7, 2024
Get rid of the is_hermitian argument for lax.qdwh. If it was known that H was also positive semi-definite, the polar decomposition would be I*H. But for indefinite H, the QDWH algorithm does not differ from the general case for Hermitian inputs.
Authored June 13, 2024
Add in the tutorial the idea to test 1 process per node and 1 process per GPU.
Authored May 30, 2024
[Pallas] Add missing trace_stop ops for jaxprs that end without a non-scoped op.
Authored June 12, 2024
jax.scipy.special.beta: deprecate x,y in favor of a,b
Authored June 10, 2024
Deprecate hashing of tracers
Authored June 13, 2024
Register beta args deprecation
Authored June 11, 2024
Adjust test tolerances for Eigh and Svd to make tests pass on all platforms.
Authored June 17, 2024
tree_all: add support for is_leaf
Authored June 10, 2024
Improve tensorstore I/O efficiency
Authored June 11, 2024
Pallas GPU no longer falls back to lax.pow for integer powers
Authored June 18, 2024
[Rollback] Bumped the minimum ml_dtypes version to 0.4.0
Authored June 12, 2024
Stop using deprecated device= argument to jax.jit in effects tests
Authored June 14, 2024
Update nightly installation instructions to reflect new jax/jaxlib dependency structure.
Authored June 14, 2024
[Mosaic GPU] Change FlashAttention implementation to support Grouped Query Attention.
Authored June 12, 2024
Add example code snippets to jax.scipy.signal.convolve2d, correlate and correlate2d docs
Authored June 10, 2024
adding doc string to right_shift
Authored June 7, 2024
Add test for QDWH with dynamic shapes.
Authored June 13, 2024

Browse Other Releases

Latest Pending Unreleased 😎
jax-v0.4.32-rc Released September 11, 2024
0 Ξ”
jax-v0.4.31 Released July 29, 2024
0 Ξ”
jax-v0.4.31-rc Released July 29, 2024
43,117 Ξ”
jax-v0.4.30 Released June 18, 2024
0 Ξ”
jax-v0.4.30-rc Released June 18, 2024
10,316 Ξ”
jax-v0.4.29 Released June 10, 2024
0 Ξ”
jax-v0.4.29-rc Released June 10, 2024
27,592 Ξ”
jax-v0.4.28 Released May 9, 2024
0 Ξ”
jax-v0.4.28-rc Released May 9, 2024
0 Ξ”
jax-v0.4.27 Released May 7, 2024
0 Ξ”

Top Contributors in jax-v0.4.30-rc

gnecula
justinjfu
a-googler
jakevdp
yashk2810
superbobry
andportnoy
junwhanahn
ayaka14732
hawkinsp

Directory Browser for jax-v0.4.30-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