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:
585,125 Diff Delta
Contributors:
439 total committers
Data confidence:
---
Commits:
5,822 impacting commits

5822 Commits in this Release

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

Stackless yashful
Authored October 2, 2024
new-sharding-docs
Authored December 2, 2025
more documentation updates; include links to code instead of inlined code
Authored March 12, 2024
Prepare to add ILP64 LAPACK support to JAX
Authored April 1, 2026
[host_callback] Remove most of the jax.experimental.host_callback module
Authored October 5, 2024
[Pallas] Add pallas distributed computation tutorial
Authored August 6, 2024
Enable building jaxlib w/ Mosaic
Authored July 20, 2023
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
[Pallas/Mosaic] Add support for nested `ref.at`
Authored December 29, 2023
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
[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
Refactor pipeline emitter API
Authored April 23, 2024
[MOSAIC] apply_vector_layout rewrite (6): arith.constant
Authored August 9, 2023
Refactor pipeline emitter API
Authored May 8, 2024
Reverts aaec4d1ef9568ac5bea7d48497fde996aafc8402
Authored April 6, 2026
[Mosaic] apply_vector_layout C++ rewrite (19): vector.transpose
Authored October 4, 2023
[pmap] Remove remaining `pmap` references from `jax`
Authored March 19, 2026
Get rid of old pallas emit_pipeline features that no one uses
Authored March 25, 2026
Move new sharding doc to 101
Authored April 3, 2026
[export] Add support for serialization and deserialization of Exported
Authored November 30, 2023
init
Authored May 14, 2024
init
Authored May 20, 2024
[shard-map] add rewrite for efficient transposition
Authored May 10, 2023
Add support for parallelizing lapack invocations on the batch dimension
Authored April 7, 2026
shard map without linear util
Authored March 9, 2026
Breaking existing tests
Authored January 13, 2026
[Mosaic] apply_vector_layout C++ rewrite: Enable apply_vector_layout_test.py
Authored October 19, 2023
[JAX] Improve handling of metadata in compilation cache
Authored April 17, 2023
migrate more internal dependencies from `jax.core` to `jax._src.core`
Authored February 11, 2023
[fixit][Pallas] Restructure shared ops tests to have one base class and per-backend subclasses
Authored March 27, 2026
Avoid imports from the public jax.* namespace in more places internally
Authored April 4, 2023
[Pallas TPU] Add matrix multiplication tutorial
Authored February 28, 2024
[Pallas MGPU] We currently make no promises about the out of bounds values of GMEM->SMEM copies
Authored March 9, 2026
[jaxprs] The next step in ClosedJaxpr simplification
Authored July 13, 2025
[Mosaic] apply_vector_layout C++ rewrite (16): vector.multi_reduction
Authored September 26, 2023
[2/n] Embed the tf.Graph into the stablehlo.custom_call
Authored April 13, 2023

Browse Other Releases

Latest Pending Unreleased 😎
0.3.25 Released November 15, 2022
8,314 Ξ”
0.3.24 Released November 4, 2022
22,228 Ξ”
0.3.23 Released October 12, 2022
0 Ξ”
0.3.22 Released October 11, 2022
22 Ξ”
0.3.21 Released September 30, 2022
0 Ξ”
0.3.20 Released September 28, 2022
0 Ξ”
jaxlib-v0.4.32 Released September 11, 2024
25,098 Ξ”
jaxlib-v0.4.31 Released July 29, 2024
49,788 Ξ”
jaxlib-v0.4.30 Released June 18, 2024
13,835 Ξ”
jaxlib-v0.4.29 Released June 10, 2024
30,491 Ξ”

Top Contributors in Latest Pending

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

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.