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

Jax: jaxlib-v0.4.26 Release

Release date:
May 18, 2024
Previous version:
jaxlib-v0.4.25 (released February 24, 2024)
Magnitude:
19,420 Diff Delta
Contributors:
64 total committers
Data confidence:
Commits:
389 impacting commits

389 Commits in this Release

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

Expose `.layout` on jax.Array. 0
Authored March 22, 2024
DOC: add key concepts doc
Authored March 21, 2024
[Take 2] Expose .layout on jax.Array. 0
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
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
Remove support for CUDA 11
Authored March 22, 2024

Browse Other Releases

Latest Pending Unreleased 😎
jaxlib-v0.4.30 Released June 18, 2024
13,835 Ξ”
jaxlib-v0.4.29 Released June 10, 2024
30,491 Ξ”
jaxlib-v0.4.28 Released May 9, 2024
0 Ξ”
jaxlib-v0.4.27 Released May 7, 2024
36,331 Ξ”
jaxlib-v0.4.26 Released May 18, 2024
19,420 Ξ”
jaxlib-v0.4.25 Released February 24, 2024
19,014 Ξ”
jaxlib-v0.4.24 Released February 5, 2024
35,114 Ξ”
jaxlib-v0.4.23 Released December 14, 2023
930 Ξ”
jaxlib-v0.4.22 Released December 13, 2023
7,552 Ξ”
jaxlib-v0.4.21 Released December 4, 2023
21,669 Ξ”

Top Contributors in jaxlib-v0.4.26

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

Directory Browser for jaxlib-v0.4.26

We haven't yet finished calculating and confirming the files and directories changed in this release. Please check back soon.