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:
615,492 Diff Delta
Contributors:
459 total committers
Data confidence:
---
Commits:
6,122 impacting commits

6122 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
remove qdd
Authored July 31, 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
Remove 'from __future__ import annotations'
Authored June 26, 2024
[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
Flattree improvements
Authored June 22, 2026
[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
FlatTree improvements
Authored June 11, 2026
Inlined all builder.create_* calls
Authored February 1, 2024
Add Mosaic lowering for Pallas
Authored July 20, 2023
[vjp3] add a separate channel of structured residuals to AD
Authored July 11, 2026
internal change
Authored July 19, 2026
[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
[pallas:mosaic_gpu] Use `emit_pipeline` for pipelining in the lowering
Authored February 25, 2025
[Mosaic GPU] Add race-checking for tensor core operations
Authored July 23, 2026
[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

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
dougalm
yashk2810
hawkinsp
apaszke
jakevdp
tlongeri
mattjj

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.