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

Jax: jax-v0.4.26-rc Release

Release date:
April 3, 2024
Previous version:
jax-v0.4.25 (released February 24, 2024)
Magnitude:
17,079 Diff Delta
Contributors:
64 total committers
Data confidence:
Commits:
392 impacting commits

392 Commits in this Release

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

Expose `.layout` on jax.Array. Also add checks in the AOT path to make sure that the input `Array`'s layout matches the layout given to `jax.jit`.
Authored March 22, 2024
DOC: add key concepts doc
Authored March 21, 2024
[Take 2] Expose .layout on jax.Array. Also add checks in the AOT path to make sure that the input Array's layout matches the layout given to jax.jit.
Authored March 25, 2024
Evaluate the correctness of JAX complex functions using mpmath as a reference
Authored March 21, 2024
[mutable-arrays] allow state effects in jit by building in run_state
Authored February 26, 2024
Reverts bed4f65438a62777ed100ecec2b0eb3f7cf87a0e
Authored March 22, 2024
[XLA:Mosaic] Use different MXU shape based on the target
Authored March 5, 2024
Reverts 0dde8f7f9607d09841ece7125dfc0773c3613fab
Authored March 27, 2024
Skip Pallas tests on GPUs earlier than Ampere (SM 8.0).
Authored March 28, 2024
Reverts 0e092a77067dbbce33cfd6d54a46e743b779919b
Authored March 22, 2024
[mutable-arrays] move MutableArray, add eager, improve tests, fix bug
Authored March 1, 2024
Add commentary and clean the pallas bidirectional collective allgather matmul test.
Authored March 26, 2024
[key reuse] improve KeyReuseSignature semantics
Authored March 8, 2024
Fixed lowering of binary ops for signed dtypes
Authored March 11, 2024
Axis names are now tracked via an effect
Authored March 4, 2024
[scan] don't traverse body jaxpr in lowering
Authored February 15, 2024
[Pallas TPU] Add support for dynamic sized (tile aligned) DMAs
Authored March 22, 2024
Refactorings to the jit implementation.
Authored March 21, 2024
[XLA:Python] Add a C++ implementation of flatten_one_level.
Authored March 21, 2024
add is_training && fix seqlen/head_dim checks
Authored March 14, 2024
Don't tree_flatten in_shardings and out_shardings each time a jit() is traced.
Authored March 21, 2024
[mutable-arrays] support closed-over mutable arrays in jit
Authored March 6, 2024
* Support automating single slice GKE TPU cluster via Job API.
Authored March 16, 2024
[key reuse] add jax.random.clone
Authored March 8, 2024
Let XLA choose in_shardings for inputs who sharding is unspecified.
Authored February 27, 2024
Allows for splitting the transpose of a scan into a scan and a map.
Authored March 28, 2024
ignore NamedAxisEffect for remat and dce purposes
Authored March 5, 2024
Remove internal references to deprecated jax.experimental.maps
Authored March 19, 2024
Convert in_shardings to physical shardings in cpp dispatch path because the same happens with prng arrays.
Authored March 6, 2024
Compute axis_index without creating an entire grid of device IDs.
Authored March 22, 2024
Remove `physical_hlo_sharding` from TyRules.
Authored March 15, 2024
Added a lowering rule for lax.sign_p and improved test coverage for binary ops
Authored March 12, 2024
[Pallas] Add stride in Pallas dynamic slice and support strided load/store.
Authored March 14, 2024
[Pallas TPU Pipeline] Add support for arbitrary manual prefetches in pipeline callbacks + compute execution.
Authored February 27, 2024
Removed unnecessary BUILD dependencies from :ops_test
Authored April 2, 2024
Include source info as ir.Locations when lowering Pallas kernels on GPU
Authored March 11, 2024
Contrain the trailing dims of prng key array to REPLICATED and keep other dims as unconstrained.
Authored February 28, 2024
Fix unnecessary memory copies between GPU and CPU when `jax2tf.call_tf()` is used.
Authored February 27, 2024
Add jax.numpy.trapezoid
Authored April 1, 2024
Disable some tests that fail on H100 in CI.
Authored March 5, 2024

Browse Other Releases

Latest Pending Unreleased 😎
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 Ξ”
jax-v0.4.27-rc Released May 7, 2024
18,638 Ξ”
jax-v0.4.26 Released April 3, 2024
0 Ξ”
jax-v0.4.26-rc Released April 3, 2024
17,079 Ξ”
jax-v0.4.25 Released February 24, 2024
0 Ξ”
jax-v0.4.25-rc Released February 24, 2024
17,651 Ξ”
jax-v0.4.24 Released February 5, 2024
0 Ξ”
jax-v0.4.24-rc Released February 5, 2024
30,244 Ξ”

Top Contributors in jax-v0.4.26-rc

yashk2810
jakevdp
mattjj
a-googler
hawkinsp
superbobry
bythew3i
pearu
gnecula
sharadmv

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