COMMITS
August 27, 2026
J
Merge pull request #40215 from ROCm:users/gulsumgudukbay/deprecate-rocm-smi-cli-upstream
jax authors committed
August 26, 2026
R
[pltpu.emit_pipeline] Allow emit_pipeline to filter no_block_spec arguments out
Robert Dyro committed
J
[typing] improve type annotations in `jaxlib.config`
Jake VanderPlas committed
Y
J
jnp.tri: use default float dtype
Jake VanderPlas committed
P
[Mosaic GPU] Add more primitives to GPU interpret mode
Paul Biberstein committed
J
Merge pull request #40173 from magaonka-amd:rocm/ci-switch-to-rocm-10.0
jax authors committed
P
Cache calls to compute_index in the pipeline emitter.
Peter Hawkins committed
J
Merge pull request #40104 from aybchan:pallas-mosaic-test-sharding
jax authors committed
J
Merge pull request #40225 from ROCm:cph-switch-pool
jax authors committed
J
Merge pull request #40236 from mattjj:push-oltpvymkryts
jax authors committed
M
[docs] highlight new matrix multiplication precision control docs
mattjj committed
B
Add `//jax/_src/pallas/mosaic_gpu:layout_rules` to wheel sources.
Benjamin Chetioui committed
A
Remove the obsolete `type_checking = "pyrefly"` attribute across `.bzl` files.
Amer Elsheikh committed
B
Increase `shard_count` to `4` for `jet_test` on GPU.
Benjamin Chetioui committed
C
[mgpu] Add missing `commit_smem`s in `mosaic_gpu_test`.
Chris Jones committed
J
T
[Mosaic] Specify scan dimension in tpu.scan
Tomás Longeri committed
C
Switch ROCm RBE to linux_x64_gpu_do_gfx950 pool
Charles Hofer committed
J
Merge pull request #40223 from mattjj:push-ryoqplqkztyu
jax authors committed
J
[typing] Use PEP 695 typevars in jaxlib.pytree
Jake VanderPlas committed
M
[docs] highlight the new (hijax-based) ffi docs
Matthew Johnson committed
July 11, 2026
T
jnp.linalg.cond: return inf rather than NaN for singular matrices
Teddy Tennant committed
August 24, 2026
M
[ROCm] Switch the pinned ROCm CI legs from 7.14 to 10.0
magaonka-amd committed
August 25, 2026
J
Merge pull request #40219 from mattjj:push-skysvottzsln
jax authors committed
M
[docs] move host-offloading content to 201/memory-spaces.md, other moves
Matthew Johnson committed
J
Merge pull request #40217 from mattjj:push-lrxvwsozvkuo
jax authors committed
G
[ROCm] Replace rocm-smi with amd-smi in CI diagnostics
gulsumgudukbay committed
J
Merge pull request #40216 from mattjj:push-zqsmnmuwooyr
jax authors committed
M
[docs] move up pallas link in sidebar, tweaks
Matthew Johnson committed