Home » Oreps » Jax » All Releases » pending_release 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
ReleasesLatest Pending
Basic Details Issues Addressed Top Contributors Directory Browser
Jax Latest Pending

Jax: Latest Pending Release

Release date:
Pending release in a future version
Previous version:
0.3.25 (released November 15, 2022)
Magnitude:
578,513 Diff Delta
Contributors:
451 total committers
Data confidence:
---
Commits:
5,982 impacting commits

5982 Commits in this Release

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

Stackless yashful
Authored October 2, 2024
Initial commit for Mosaic GPU
Authored April 15, 2024
more documentation updates; include links to code instead of inlined code
Authored March 12, 2024
Add Mosaic to Jaxlib and expose bindings in `jax.experimental.mosaic`
Authored July 14, 2023
Initial commit for Mosaic GPU
Authored April 16, 2024
[XLA:MGPU] Delete C++ implementation of core primitives
Authored April 24, 2026
Enable building jaxlib w/ Mosaic
Authored July 20, 2023
[host_callback] Remove most of the jax.experimental.host_callback module
Authored October 5, 2024
Add new scripts and configs for building JAX artifacts
Authored September 16, 2024
Changed lower_jaxpr_to_triton_ir to use ir.Value instead of triton.compat APIs
Authored February 9, 2024
Cherry picked for testing
Authored September 12, 2025
[Pallas] Add pallas distributed computation tutorial
Authored August 6, 2024
Port Triangular Solve to XLA's FFI
Authored June 3, 2024
Port Householder Product to XLA's FFI
Authored June 3, 2024
Regenerate python stubs
Authored May 20, 2024
[Mosaic] apply_vector_layout C++ rewrite (8): tpu.gather, tpu.iota, tpu.trace
Authored August 21, 2023
[Mosaic] apply_vector_layout C++ rewrite (10): vector.extract_strided_slice
Authored September 6, 2023
[Pallas/Mosaic] Add support for nested `ref.at`
Authored December 29, 2023
[MOSAIC] apply_vector_layout C++ rewrite (7): tpu load and store
Authored August 9, 2023
[Mosaic] apply_vector_layout C++ rewrite (9): tpu.repeat
Authored September 5, 2023
[MOSAIC] apply_vector_layout rewrite (6): arith.constant
Authored August 9, 2023
Use vars instead of argument-list indices in effect types
Authored June 9, 2026
Refactor pipeline emitter API
Authored April 23, 2024
[Mosaic] apply_vector_layout C++ rewrite (19): vector.transpose
Authored October 4, 2023
Inlined all builder.create_* calls
Authored February 1, 2024
Add Mosaic lowering for Pallas
Authored July 20, 2023
[Pallas TPU] Add flash attention tutorial
Authored March 6, 2024
Bump minimum jaxlib version to 0.4.27
Authored May 7, 2024
[export] Add support for serialization and deserialization of Exported
Authored November 30, 2023
[jax2tf] Clean up pyrefly typing errors
Authored May 22, 2026
[pallas:mosaic_gpu] Use `emit_pipeline` for pipelining in the lowering
Authored February 25, 2025
[Pallas] Refactor indexing primitives to use NDIndexer abstraction
Authored December 6, 2023
[Mosaic TPU] Move C++ tests out of the JAX repository
Authored May 6, 2026
[shard-map] add rewrite for efficient transposition
Authored May 10, 2023
[Mosaic] apply_vector_layout C++ rewrite: Enable apply_vector_layout_test.py
Authored October 19, 2023
shard map without linear util
Authored March 9, 2026
add shmap-transpose JEP
Authored August 16, 2023
init
Authored May 14, 2024
[Mosaic GPU] Turn on equations-based layout inference
Authored August 14, 2025
init
Authored May 20, 2024

Browse Other Releases

Latest Pending Unreleased 😎
0.3.25 Released November 15, 2022
6,770 Ξ”
0.3.24 Released November 4, 2022
20,656 Ξ”
0.3.23 Released October 12, 2022
0 Ξ”
0.3.22 Released October 11, 2022
16 Ξ”
0.3.21 Released September 30, 2022
0 Ξ”
0.3.20 Released September 28, 2022
0 Ξ”
jaxlib-v0.4.32 Released September 11, 2024
22,228 Ξ”
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 Ξ”

Top Contributors in Latest Pending

a-googler
gnecula
superbobry
hawkinsp
yashk2810
dougalm
jakevdp
apaszke
tlongeri
sharadmv

Directory Browser for Latest Pending

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