Pull Requests ROCm/jax - SemanticDiff

Abort detection ci workflow gulsum

#688 opened 2026-02-04 05:43 by gulsumgudukbay

Enable lax backend scipy tests on ROCm GPUs open-upstream cherry-pick-candidate

#687 opened 2026-02-03 20:54 by magaonka-amd

Enable lax backend scipy tests on ROCm GPUs open-upstream cherry-pick-candidate

#686 opened 2026-02-03 20:53 by magaonka-amd

Enable lax backend scipy tests on ROCm GPUs open-upstream cherry-pick-candidate

#685 opened 2026-02-03 20:51 by magaonka-amd

Add ROCm encoding for test_struct_encoding_determinism open-upstream cherry-pick-candidate

#684 opened 2026-02-03 15:47 by AratiGanesh

Add ROCm encoding for test_struct_encoding_determinism open-upstream cherry-pick-candidate

#683 opened 2026-02-03 15:44 by AratiGanesh

Add ROCm-specific binary encoding for test_struct_encoding_determinism open-upstream cherry-pick-candidate

#679 opened 2026-01-30 22:07 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

Test pytest abort plugin

#670 opened 2026-01-28 23:32 by gulsumgudukbay

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