Pull Requests ROCm/jax - SemanticDiff

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

Add script for running bazel tests on ROCm

#632 opened 2026-01-21 19:12 by charleshofer

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

[0.6.0-UT] Adjust tolerance for reciprocal in unary primitives test

#518 opened 2025-07-25 21:31 by AratiGanesh

CI: 06/20/25 upstream sync

#477 opened 2025-06-20 06:02 by rocm-repo-management-api-2[bot]

Add rocm perf yml file

#418 opened 2025-05-12 15:43 by Ruturaj4

Add linear algebra tests to our CI workflow

#388 opened 2025-04-28 14:03 by charleshofer

gfx950 platform support

#381 opened 2025-04-23 17:22 by zahiqbal

Fix CI cleanup race condition

#349 opened 2025-04-08 19:40 by charleshofer

Update latest xla hash for JAX branch 0.5.0

#337 opened 2025-04-02 23:32 by Ruturaj4

Find pyversion at build time 0.5.0

#328 opened 2025-03-28 19:57 by mrodden

[rocm-main]: Canonicalize python build versions

#321 opened 2025-03-26 22:31 by mrodden

Add support for ROCm wheel based install

#305 opened 2025-03-21 19:47 by mrodden

Update XLA Commit hash

#266 opened 2025-03-07 16:31 by JehandadKhan

[ROCm] Implement RNN support open-upstream

#171 opened 2024-12-02 16:57 by Ruturaj4