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:
680,291 Diff Delta
Contributors:
457 total committers
Data confidence:
---
Commits:
6,278 impacting commits

6278 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
Port Eigenvalue Decompositions to XLA's FFI
Authored June 3, 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
Add new scripts and configs for running JAX tests
Authored September 16, 2024
[XLA:MGPU] Delete C++ implementation of core primitives
Authored April 24, 2026
aiter mha kernels (ASM+CK) integration (#747)
Authored April 26, 2026
Squashed commit of the following:
Authored September 18, 2024
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
Cleanup: access tree utilities via jax.tree.*
Authored February 23, 2024
Cherry picked for testing
Authored September 12, 2025
Remove 'from __future__ import annotations'
Authored June 26, 2024
Checkin cutlass experimental package
Authored July 1, 2025
Jaxpr pass to resolve extended dtypes
Authored September 18, 2024
[Pallas] Add pallas distributed computation tutorial
Authored August 6, 2024
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
Regenerate python stubs
Authored May 20, 2024
[Mosaic] apply_vector_layout C++ rewrite: Fix check for defining layout in disassemble
Authored September 27, 2023
Remove deprecated is_cloud_tpu_at_least version guards
Authored June 15, 2026
[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
Use hermetic python in TSL/XLA/JAX
Authored July 18, 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
updated docs; dump custom_partitioning HLO
Authored March 11, 2024
[MOSAIC] apply_vector_layout rewrite (6): arith.constant
Authored August 9, 2023
[Mosaic] apply_vector_layout C++ rewrite: Handle elementwise ops by checking for the Elementwise trait and using the generic Operation interface, without templates
Authored October 4, 2023
Use vars instead of argument-list indices in effect types
Authored June 9, 2026
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

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

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