Pull Requests ROCm/jax - SemanticDiff

[ROCm] Deselect crashing FusedAttention bwd8 + add pytest-timeout (ROCM-25626)

#798 opened 2026-06-09 15:28 by srinivamd

[ROCm] Skip 6 Pallas FusedAttentionTest variants exceeding gfx942 LDS limit

#797 opened 2026-06-09 10:49 by srinivamd

[ROCm] xdist: select GPU from runner-allocated ROCR slice

#793 opened 2026-06-05 21:58 by mminutoli

ci: deselect PGLE/FDO profiling tests on ROCm (ROCM-25581) [v0.9.2]

#790 opened 2026-06-03 15:33 by srinivamd

Disable the cron trigger for the cont. wheel tests workflow

#789 opened 2026-06-03 15:32 by psanal35

ci: deselect PGLE/FDO profiling tests on ROCm (ROCM-25581)

#788 opened 2026-06-03 15:31 by srinivamd

Cherry-pick ROCm fixes from PR #740 to rocm-jaxlib-v0.8.2

#786 opened 2026-06-02 14:37 by srinivamd

Cherry-pick ROCm fixes from PR #740 to rocm-jaxlib-v0.9.0

#785 opened 2026-06-02 14:36 by srinivamd

[ROCm] Skip TF32 dot algorithm tests on Navi/RDNA GPUs

#783 opened 2026-06-01 04:24 by srinivamd

Add docker image to use hermetic llvm and sysroot

#782 opened 2026-05-29 20:13 by hsharsha

Bundle specific and generic arch build targets together.

#780 opened 2026-05-20 19:53 by tsrw2048

Fix env var leak in xla_bridge_test register_plugin tests

#775 opened 2026-05-14 18:38 by tsrw2048

Skip ann_test test_pmap on ROCm due to IndivisibleError

#773 opened 2026-05-14 14:31 by AratiGanesh

jaxlib/gpu/rnn_kernels.cc: remove premature input_tensor_desc destruction (ROCM-21859)

#765 opened 2026-04-30 14:23 by srinivamd

[ROCm] Skip core_test reduction autotuner failures on Navi/RDNA GPUs (gfx11xx)

#758 opened 2026-04-28 22:51 by CSkmd

[ROCm]---Theory--- force ROCm xdist worker GPU env override

#749 opened 2026-04-10 03:44 by magaonka-amd

port rocm/jax 0.9.2 cherrypicks to amd-main branch (#740)

#741 opened 2026-03-23 22:21 by magaonka-amd

Fix: Scaled Matmul at rocm 9.0

#739 opened 2026-03-17 23:21 by shurale-nkn

Fix: Scaled Matmul at rocm 9.1

#738 opened 2026-03-17 23:21 by shurale-nkn

Add ROCm gesdd SVD backend, export back-compat data, and relaxed GPU … enhancement

#734 opened 2026-03-06 16:41 by cj401-amd

Add ROCm gesdd SVD backend, export back-compat data, and relaxed GPU … enhancement

#733 opened 2026-03-06 16:36 by cj401-amd

update test

#732 opened 2026-03-05 23:13 by kiran-thumma

update test

#731 opened 2026-03-05 23:11 by kiran-thumma

Remove input_tensor_desc destruction in RNN kernels cherry-pick-candidate

#729 opened 2026-03-05 17:13 by magaonka-amd

Fix HIP memory leaks in RNN kernels

#728 opened 2026-03-05 16:25 by magaonka-amd

Fix HIP memory leaks in RNN kernels

#727 opened 2026-03-05 16:20 by magaonka-amd

Skip test_pmap on ROCm due to IndivisibleError with new pmap SPMD tiling

#725 opened 2026-03-04 20:47 by AratiGanesh

[Testing do not merge] Add ROCm wheel build and test pipeline to continuous CI

#723 opened 2026-03-04 14:02 by alekstheod

Fix batched Cholesky OOM on ROCm by bypassing hipSolver's external hipMalloc

#717 opened 2026-02-26 10:19 by FlemingH

Improve ROCm pytest results handling

#716 opened 2026-02-25 17:13 by psanal35