Pull Requests ROCm/jax - SemanticDiff

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

fix poor performance of slogdet when input dimension n=1 or 2 enhancement cherry-pick-candidate

#710 opened 2026-02-10 22:03 by cj401-amd

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

#701 opened 2026-02-09 21:13 by AratiGanesh

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

#699 opened 2026-02-09 20:30 by AratiGanesh

Enable scaled matmul tests on ROCm

#678 opened 2026-01-30 20:49 by magaonka-amd

Rocm jaxlib v0.9.0

#671 opened 2026-01-28 23:56 by Ruturaj4

update with ROCm profiling tests with kernel_details

#664 opened 2026-01-28 14:34 by cj401-amd

remove skipTest associated to hipsparse transpose

#661 opened 2026-01-27 22:37 by kliegeois

Support wheels as build deps

#638 opened 2026-01-22 16:41 by alekstheod

Use xla from rocm-jaxlib-v0.8.0 release branch

#599 opened 2026-01-14 14:44 by alekstheod

Add CI workflow for Pytest ROCm

#594 opened 2026-01-12 18:17 by mminutoli

Upstream ci download jax rocm wheels

#592 opened 2026-01-10 00:57 by mminutoli

Adding run_pytest_rocm.sh

#591 opened 2026-01-09 02:35 by gulsumgudukbay

update a test for checking zero ROCm GPU event open-upstream cherry-pick-candidate

#585 opened 2025-12-18 23:36 by cj401-amd

Use plain bazel to test jax, use hermetic rocm dependency

#584 opened 2025-12-18 12:33 by alekstheod

Fix/pallas tests shared memory

#581 opened 2025-12-15 15:39 by AratiGanesh

Create rocm-test-requirements.txt

#579 opened 2025-12-12 12:00 by jvadakat

[0.6.0-UT] Fix Test Shard

#519 opened 2025-07-29 20:41 by AratiGanesh