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:
13,835 Diff Delta
Contributors:
29 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] Add documentation for jax.export
Authored May 31, 2024
[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
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. 0
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. 0
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
Update nightly installation instructions to reflect new jax/jaxlib dependency structure
Authored June 14, 2024
Stop using deprecated device= argument to jax.jit in effects tests
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
35,570 Ξ”
jax-v0.4.31 Released July 29, 2024
0 Ξ”
jax-v0.4.31-rc Released July 29, 2024
51,898 Ξ”
jax-v0.4.30 Released June 18, 2024
0 Ξ”
jax-v0.4.30-rc Released June 18, 2024
13,835 Ξ”
jax-v0.4.29 Released June 10, 2024
0 Ξ”
jax-v0.4.29-rc Released June 10, 2024
30,630 Ξ”
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
junwhanahn
ayaka14732
hawkinsp
rajasekharporeddy

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.