Pull Requests ROCm/jax - SemanticDiff

Backport approx_tanh fix (cherry-pick jax-ml/jax#34598)

#750 by AratiGanesh was merged 2026-04-20 18:46

Temporarily disable the cron trigger for the cont. wheel tests workflow

#748 by psanal35 was merged 2026-04-08 13:58

aiter mha kernels (ASM+CK) integration

#747 by zahiqbal was merged 2026-04-26 22:53

[ROCm] Cherry-pick: Add ROCm xdist device pinning in conftest.py

#744 by magaonka-amd was merged 2026-04-02 15:42

Fixes for CI Nightly Bazel Workflow

#743 by tsrw2048 was closed 2026-05-08 16:20

Fix deprecated mpmath.ctx_mp.mpnumeric import for mpmath >= 1.4.0

#742 by AratiGanesh was merged 2026-04-20 18:46

Cherry-pick fixes for JAX 0.9.2 release from JAX 0.9.1

#740 by magaonka-amd was merged 2026-03-23 22:13

fix: add rocm_sysdeps/lib to wheel RUNPATH

#737 by WBobby was merged 2026-03-17 22:40

Fix: Scaled Matmul at rocm

#736 by shurale-nkn was closed 2026-03-17 17:07

Release Prep JAX 0.9.1

#735 by gulsumgudukbay was merged 2026-03-20 16:14

Revert "Fix HIP memory leaks in RNN kernels"

#730 by magaonka-amd was merged 2026-05-01 19:59

Fix HIP memory leaks in RNN kernels

#726 by magaonka-amd was merged 2026-03-05 16:00

[Test do not merge] Implement periodic build wheels pipeline rocm internal v3

#724 by alekstheod was closed 2026-03-04 14:55

[Test do not merge] Testing build wheels

#722 by alekstheod was closed 2026-03-04 14:03

Implement periodic build wheels pipeline rocm internal

#721 by alekstheod was closed 2026-03-04 11:19

Phambinh/skip fused attention navi lds limit

#720 by phambinhfin was merged 2026-03-04 12:13

Remove 'var': ['mean'] from unsupported_params in signature test

#719 by phambinhfin was merged 2026-02-26 16:06

Add ROCm platform to export memory_space tests

#715 by phambinhfin was merged 2026-02-24 16:20

Implement Mosaic GPU detection and Auto-Skips open-upstream

#714 by gulsumgudukbay was closed 2026-05-15 14:00

The workflow is missing the halt-for-connection argument.

#713 by mminutoli was closed 2026-02-20 17:52

Abort-Detection: Fix halt-for-connection input

#712 by gulsumgudukbay was merged 2026-02-24 15:59

Fix abort workflow

#711 by gulsumgudukbay was closed 2026-02-20 17:46

Skip test_batch_axis_sharding_jvp13 test 0.9.0 open-upstream cherry-pick-candidate

#709 by AratiGanesh was merged 2026-02-11 01:30

Skip test_batch_axis_sharding_jvp13 test 0.8.0 open-upstream cherry-pick-candidate

#708 by AratiGanesh was merged 2026-02-16 16:10

Skip test_batch_axis_sharding_jvp13 test on 0.8.2 open-upstream cherry-pick-candidate

#707 by AratiGanesh was merged 2026-02-16 16:11

Backporting `device test` unit tests for ROCm from JAX v0.9.0 to v0.8.2. open-upstream cherry-pick-candidate

#706 by tsrw2048 was merged 2026-02-16 16:44

Add `device test` unit tests for ROCm (JAX v0.9.0) open-upstream cherry-pick-candidate

#705 by tsrw2048 was merged 2026-02-10 19:42

Skip test_tridiagonal_solve_grad test 0.8.2 open-upstream cherry-pick-candidate

#704 by AratiGanesh was merged 2026-02-16 16:11

Skip test_tridiagonal_solve_grad test 0.9.0 open-upstream cherry-pick-candidate

#703 by AratiGanesh was merged 2026-02-11 01:28

Update skip message version from 0.8.0 to 0.9.0 for test_is_finite on…

#702 by phambinhfin was merged 2026-02-12 13:38