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:
724,051 Diff Delta
Contributors:
444 total committers
Data confidence:
---
Commits:
5,975 impacting commits

5975 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
Initial commit for Mosaic GPU
Authored April 15, 2024
Initial commit for Mosaic GPU
Authored April 16, 2024
Port Eigenvalue Decompositions to XLA's FFI
Authored June 3, 2024
Add Mosaic to Jaxlib and expose bindings in `jax.experimental.mosaic`
Authored July 14, 2023
Add new scripts and configs for running JAX tests
Authored September 16, 2024
Cherry picked for testing
Authored September 12, 2025
Reverts 9cad5da07c68ac26277679236c46345906f8c866
Authored April 2, 2026
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
Squashed commit of the following:
Authored September 18, 2024
[Pallas] Add pallas distributed computation tutorial
Authored August 6, 2024
Enable building jaxlib w/ Mosaic
Authored July 20, 2023
Cleanup: access tree utilities via jax.tree.*
Authored February 23, 2024
Add new scripts and configs for building JAX artifacts
Authored September 16, 2024
Remove 'from __future__ import annotations'
Authored June 26, 2024
Jaxpr pass to resolve extended dtypes
Authored September 18, 2024
[pmap] Remove the config state `jax_pmap_shmap_merge`
Authored March 19, 2026
Changed lower_jaxpr_to_triton_ir to use ir.Value instead of triton.compat APIs
Authored February 9, 2024
Checkin cutlass experimental package
Authored July 1, 2025
Dropped the Triton* prefix from class names in triton/lowering.py
Authored February 12, 2024
Port Triangular Solve to XLA's FFI
Authored June 3, 2024
[Mosaic] apply_vector_layout C++ rewrite (13): scf.if
Authored September 7, 2023
Port Householder Product to XLA's FFI
Authored June 3, 2024
[Mosaic] apply_vector_layout C++ rewrite (11): vector.broadcast
Authored September 6, 2023
[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: Fix check for defining layout in disassemble
Authored September 27, 2023
[Mosaic] apply_vector_layout C++ rewrite (8): tpu.gather, tpu.iota, tpu.trace
Authored August 21, 2023
Use hermetic python in TSL/XLA/JAX
Authored July 18, 2023
[Mosaic] apply_vector_layout C++ rewrite (10): vector.extract_strided_slice
Authored September 6, 2023
[Pallas TPU] Add flash attention tutorial
Authored March 6, 2024
Resolve TypeError and test failures in state_test_cpu by updating trace_to_jaxpr call sites to comply with the new API (strictly requiring debug_info)
Authored March 24, 2026
[shape_poly] Copy many of the jax2tf/shape_poly_test outside of jax2tf
Authored November 15, 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

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

a-googler
gnecula
hawkinsp
superbobry
tlongeri
jakevdp
yashk2810
dougalm
apaszke
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.

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