Home » Oreps » Jax » All Releases » jaxlib-v0.1.33 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
Releasesjaxlib-v0.1.33
Basic Details Issues Addressed Top Contributors Directory Browser
Jax jaxlib-v0.1.33

Jax: jaxlib-v0.1.33 Release

Release date:
January 21, 2026
Previous version:
jax-v0.9.2 (released March 18, 2026)
Magnitude:
12,194 Diff Delta
Contributors:
28 total committers
Data confidence:
Commits:
145 impacting commits

145 Commits in this Release

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

internal change
Authored March 12, 2026
[Take 2] shard map without linear util
Authored March 13, 2026
[Pallas/TPU] Add tpu_info.get_tpu_info_for_chip
Authored February 26, 2026
Add int1/uint1 dtypes
Authored February 26, 2026
[pyrefly] type check jax/experimental/pallas
Authored March 6, 2026
Add workflow for build wheels and running bazel tests on ROCm
Authored January 21, 2026
[pyrefly] add type checking for jax/experimental/sparse
Authored March 10, 2026
[Pallas][TPU kernel interpreter] Consistently use `pltu.HBM` to refer to HBM in the interpreter
Authored February 27, 2026
[NFC] Remove and modify some out-of-date test skips
Authored March 2, 2026
[Mosaic:GPU] Re-use the barrier buffers between executions to prevent situation when the buffer were released before the end of cross-device barrier
Authored February 27, 2026
[Pallas][Mosaic TPU] More complete 1D block support in Pallas/Mosaic
Authored February 28, 2026
[export] Remove code to serialize HloSharding
Authored March 3, 2026
[pyrefly] type-check jax.experimental/array_serialization & colocated_python
Authored March 10, 2026
[hijax] cleaner scan inc_rank/dec_rank api
Authored February 26, 2026
Add `HloShardingV3 -> JAX NamedSharding` conversion, used for shardings returned via compiled XLA executable back to JAX
Authored March 9, 2026
Show detailed mismatch key paths in `flatten_axes()` tree prefix errors
Authored March 7, 2026
[Pallas/Mosaic GPU] Turn on `gpu_pallas_distributed_test`s for warpgroup semantics
Authored March 13, 2026
scipy.linalg: add hankel special matrix
Authored February 19, 2026
[maint] clean up array class overrides
Authored March 12, 2026
[typing] miscellaneous pyrefly typing fixes
Authored February 26, 2026
[Mosaic GPU][NFC] Remove dead `constraints.BroadcastInDim`
Authored March 2, 2026
Add C addition to hopper matmul kernel
Authored March 2, 2026
[pyrefly] fix errors in jax/interpreters/ad
Authored February 27, 2026
[doc] update AI code policy
Authored March 9, 2026
[Mosaic GPU]Β Add `a_sparse_metadata` argument to `tcgen05_mma` dialect definition
Authored March 6, 2026
[Pallas/Mosaic GPU] Enable lowering `semaphore_{read,signal,wait}_p` using warpgroup semantics
Authored March 10, 2026
[Mosaic] Simplify GetCoreTypeOfParentOp
Authored February 28, 2026
Bugfix for colocated python with `use_weakrefs_at_backend=True`
Authored March 12, 2026
Tracer is now generic parameterized by Trace
Authored March 13, 2026
Tie LinearizeTrace.tag to tangent_trace.tag
Authored February 26, 2026
[Mosaic GPU] Add support for reducing strided and splat reduce layout constaints
Authored March 10, 2026
[jaxlib] Ported normalize_stubs.sh to Python
Authored February 26, 2026
Fix a bug in sort_key_val for negative integers, since SC sort treats as unsigned
Authored March 13, 2026
[mlir] Fixed a few type errors in MLIR usages
Authored March 12, 2026
[Pallas] Add more test cases for folding last two dims reshape
Authored March 10, 2026
[Pallas/Mosaic GPU] Fix misuse of `self.{kernel,pallas_call}` in `mosaic_gpu_test.py`
Authored March 13, 2026
[pyrefly] fix typing errors in jax/_src/array.py
Authored March 2, 2026
Skip tests that are failing at libtpu=0.0.35, but passing at libtpu-0.0.37.dev20260224
Authored February 27, 2026
[Pallas] Add buffering mode to pltpu that enables non-immediate revisit
Authored March 13, 2026
[Mosaic GPU] Make tests involving global state order-independent
Authored March 16, 2026

Browse Other Releases

Latest Pending Unreleased 😎
jaxlib-v0.1.39 Released February 14, 2020
0 Ξ”
jaxlib-v0.1.38 Released January 29, 2020
0 Ξ”
jaxlib-v0.1.37 Released December 6, 2019
0 Ξ”
jaxlib-v0.1.36 Released November 22, 2019
0 Ξ”
jaxlib-v0.1.33 Released January 21, 2026
12,194 Ξ”
jax-v0.9.2 Released March 18, 2026
15,566 Ξ”
jax-v0.9.1 Released February 28, 2026
19,549 Ξ”
jax-v0.9.0.1 Released February 3, 2026
0 Ξ”
jax-v0.9.0 Released January 20, 2026
0 Ξ”
jax-v0.8.3 Released January 28, 2026
28 Ξ”

Top Contributors in jaxlib-v0.1.33

a-googler
jakevdp
dougalm
bchetioui
danielsuo
yueshengys
WindQAQ
allanrenucci
charleshofer
mattjj

Directory Browser for jaxlib-v0.1.33

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

Loading File Browser...