Home » Oreps » Jax » All Releases » jax-v0.10.2 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
Releasesjax-v0.10.2
Basic Details Issues Addressed Top Contributors Directory Browser Release Notes
Jax jax-v0.10.2

Jax: jax-v0.10.2 Release

Release date:
June 17, 2026
Previous version:
jax-v0.10.1 (released May 18, 2026)
Magnitude:
28,770 Diff Delta
Contributors:
49 total committers
Data confidence:
Commits:
456 impacting commits

456 Commits in this Release

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

[jax2tf] Clean up pyrefly typing errors
Authored May 22, 2026
Remove stale version tests
Authored May 22, 2026
Use vars instead of de bruijn indices in effects
Authored June 12, 2026
Move dtypes.canonicalize_value to C++
Authored June 4, 2026
Stop using linear util in AD
Authored May 22, 2026
[pallas:mgpu] Aligned the API of `plgpu.kernel` with `pl.kernel`
Authored June 10, 2026
[state] Use `discharge_state2` everywhere and drop the 2
Authored June 5, 2026
Reverts aa296d2a281ded40c47b68cf695432d6e0c793ad
Authored June 11, 2026
Add philox2x32 PRNG impl
Authored May 15, 2026
jax: replace trace_to_jaxpr_dynamic with trace_to_jaxpr in state and interpreters
Authored May 15, 2026
[Pallas] In GPU Interpret Mode, replace ordered IO callbacks with unordered
Authored June 6, 2026
put ct avals on GradAccums (and UndefinedPrimals), not primal avals
Authored June 3, 2026
Use pytrees for flattening and unflattening in MLIR lowering
Authored May 18, 2026
[Pallas] Improve error handling in TPU Interpret Mode to avoid hangs
Authored May 28, 2026
Reverts 0af318d7b296a6822f610a547a8b97e5b45eb083
Authored May 20, 2026
[Pallas MGPU] Support wgmma ops and copy_smem_to_gmem in GPU Interpret Mode
Authored June 12, 2026
Reverts e5892751aa7eb93dc603404fa6d184f039bef14b
Authored May 18, 2026
Add threefry4x32 PRNG impl
Authored May 19, 2026
Refactor `jax/_src/random/`
Authored May 18, 2026
Delete not_mapped and replace it with None
Authored May 29, 2026
Roll back PR #36850. 0
Authored June 8, 2026
[NFC] Fix weird formatting in `mosaic_gpu_test.py`
Authored June 10, 2026
[Pallas][GPU kernel interpreter] Properly support `collective_axes` for allocations made by `run_scoped`
Authored June 9, 2026
[Mosaic GPU][NFC] Refactor `Constraint` to make sure we handle all `Constant`s
Authored May 18, 2026
[Mosaic GPU][NFC] Split swizzle and memref transform extraction from transform attributes
Authored May 18, 2026
[pallas:mgpu] Migrated our main test suite to use `plgpu.kernel`
Authored May 20, 2026
Reverts 6a4f44085525c2f72ef291f19453b2e000a299a8
Authored May 16, 2026
Update xla_transform_test to unregister callbacks after each unit test
Authored June 11, 2026
[Pallas:MGPU] Support batch dimension for TMEM references
Authored June 11, 2026
[state] Use `discharge_state2` everywhere and drop the 2
Authored June 5, 2026
Error out if `*pspec` or `pspec[...]` is used when unreduced/reduced is non-empty
Authored May 28, 2026
Clean up GitHub Actions job names for presubmits
Authored June 12, 2026
Support tiling and swizzle transforms in GPU Interpret Mode
Authored June 8, 2026
[Pallas/Mosaic GPU][Docs] Add hyperlinks to various API bits throughout the reference
Authored June 11, 2026
Small optimizations to stablehlo lowering:
Authored May 15, 2026
In Primitive.bind, canonicalize first before calling typeof and fuse the loops that compute them
Authored June 6, 2026
Add experimental jax control dependency API
Authored May 15, 2026
Add __slots__ to dataclasses in jax
Authored June 2, 2026
jax.make_mesh: re-add multi-slice error on GPU
Authored May 19, 2026
Replace references to xla_client._xla with jax._src.lib._jax
Authored June 2, 2026

Browse Other Releases

Latest Pending Unreleased 😎
jaxlib-v0.1.37 Released December 6, 2019
0 Ξ”
jaxlib-v0.1.36 Released November 22, 2019
0 Ξ”
jaxlib-v0.1.33 Released July 16, 2026
406,061 Ξ”
jax-v0.11.0 Released July 16, 2026
27,218 Ξ”
jax-v0.10.2 Released June 17, 2026
28,770 Ξ”
jax-v0.10.1 Released May 18, 2026
31,115 Ξ”
jax-v0.10.0 Released April 16, 2026
62,274 Ξ”
jax-v0.9.2 Released March 18, 2026
14,075 Ξ”
jax-v0.9.1 Released February 28, 2026
17,664 Ξ”
jax-v0.9.0.1 Released February 3, 2026
0 Ξ”

Top Contributors in jax-v0.10.2

hawkinsp
superbobry
yashk2810
a-googler
dougalm
gnecula
jburnim
allanrenucci
jakevdp
bchetioui

Directory Browser for jax-v0.10.2

All files are compared to previous version, jax-v0.10.1. Click here to browse diffs between other versions.

Loading File Browser...

Release Notes Published

  • New features
    • Added jax.scipy.linalg.invhilbert for the closed-form inverse of the Hilbert matrix ({jax-issue}#10144).
    • Added jax.scipy.linalg.invpascal for the inverse of the Pascal matrix (#10144).
    • Added jax.scipy.linalg.fiedler_companion for constructing the pentadiagonal Fiedler companion matrix of a polynomial (#10144).
    • Added jax.ShapeDtypeStruct.like -- a shortcut for constructing a jax.ShapeDtypeStruct from an object with shape and dtype attributes.
  • We have made a number of improvements to JAX tracing performance.
GitClear uses cookies to ensure you get the best experience on our website. Learn more about our cookie policy