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:
16,029 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.

setup ci for jax-metal plugin
Authored February 23, 2024
Enable Creating Device Mesh with Physical Axes Splits
Authored March 21, 2024
Expose `.layout` on jax.Array. 0
Authored March 22, 2024
[XLA:Mosaic] Use different MXU shape based on the target
Authored March 5, 2024
DOC: add key concepts doc
Authored March 21, 2024
Reverts 0dde8f7f9607d09841ece7125dfc0773c3613fab
Authored March 27, 2024
Evaluate the correctness of JAX complex functions using mpmath as a reference
Authored March 21, 2024
Reverts bed4f65438a62777ed100ecec2b0eb3f7cf87a0e
Authored March 22, 2024
[mutable-arrays] move MutableArray, add eager, improve tests, fix bug
Authored March 1, 2024
[Take 2] Expose .layout on jax.Array. 0
Authored March 25, 2024
[mutable-arrays] allow state effects in jit by building in run_state
Authored February 26, 2024
Fixed lowering of binary ops for signed dtypes
Authored March 11, 2024
[key reuse] improve KeyReuseSignature semantics
Authored March 8, 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
Reverts 0e092a77067dbbce33cfd6d54a46e743b779919b
Authored March 22, 2024
[xmap-removal] remove reduce_axes from grad / vjp / backward_pass
Authored February 25, 2024
Skip Pallas tests on GPUs earlier than Ampere (SM 8.0)
Authored March 28, 2024
Refactorings to the jit implementation
Authored March 21, 2024
Add commentary and clean the pallas bidirectional collective allgather matmul test
Authored March 26, 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
* Support automating single slice GKE TPU cluster via Job API
Authored March 16, 2024
[Pallas TPU] Add support for dynamic sized (tile aligned) DMAs
Authored March 22, 2024
[mutable-arrays] support closed-over mutable arrays in jit
Authored March 6, 2024
Compute axis_index without creating an entire grid of device IDs
Authored March 22, 2024
Let XLA choose in_shardings for inputs who sharding is unspecified
Authored February 27, 2024
ignore NamedAxisEffect for remat and dce purposes
Authored March 5, 2024
Allows for splitting the transpose of a scan into a scan and a map
Authored March 28, 2024
Remove `physical_hlo_sharding` from TyRules
Authored March 15, 2024
Convert in_shardings to physical shardings in cpp dispatch path because the same happens with prng arrays
Authored March 6, 2024
[Pallas] Add stride in Pallas dynamic slice and support strided load/store
Authored March 14, 2024
[key reuse] add jax.random.clone
Authored March 8, 2024
[Pallas TPU Pipeline] Add support for arbitrary manual prefetches in pipeline callbacks + compute execution
Authored February 27, 2024
Include source info as ir.Locations when lowering Pallas kernels on GPU
Authored March 11, 2024
Remove internal references to deprecated jax.experimental.maps
Authored March 19, 2024
Contrain the trailing dims of prng key array to REPLICATED and keep other dims as unconstrained
Authored February 28, 2024
Add a fast path for Python scalars to shaped_abstractify
Authored March 22, 2024
[key reuse] print information about key reuse location
Authored March 21, 2024
[callback] Improve caching effectiveness in presence of callbacks
Authored April 1, 2024

Browse Other Releases

Latest Pending Unreleased 😎
jaxlib-v0.4.31 Released July 29, 2024
43,313 Ξ”
jaxlib-v0.4.30 Released June 18, 2024
11,389 Ξ”
jaxlib-v0.4.29 Released June 10, 2024
26,270 Ξ”
jaxlib-v0.4.28 Released May 9, 2024
5,052 Ξ”
jaxlib-v0.4.27 Released May 7, 2024
30,921 Ξ”
jaxlib-v0.4.26 Released May 18, 2024
16,029 Ξ”
jaxlib-v0.4.25 Released February 24, 2024
17,354 Ξ”
jaxlib-v0.4.24 Released February 5, 2024
30,896 Ξ”
jaxlib-v0.4.23 Released December 14, 2023
704 Ξ”
jaxlib-v0.4.22 Released December 13, 2023
6,291 Ξ”

Top Contributors in jaxlib-v0.4.26

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

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.

GitClear uses cookies to ensure you get the best experience on our website. Learn more about our cookie policy