ROCm/jax

Pull Requests Commits
Open Closed
Skip test_batch_axis_sharding_jvp on ROCm open-upstream cherry-pick-candidate
#654 opened 2026-01-23 21:55 by AratiGanesh
Support wheels as build deps
#638 opened 2026-01-22 16:41 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
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
add rocprofiler_configure as a global
#479 opened 2025-06-20 22:30 by cj401-amd
add rocprofiler_configure as a global
#478 opened 2025-06-20 22:18 by cj401-amd
Add rocm perf yml file
#418 opened 2025-05-12 15:43 by Ruturaj4
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
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
Update http to https in amd artifactory url. open-upstream
#164 opened 2024-11-26 22:37 by Ruturaj4
[ROCm] Add argument to override base docker in dockerfile open-upstream
#163 opened 2024-11-26 22:28 by Ruturaj4
[ROCm] ci build and dockerfile changes open-upstream
#161 opened 2024-11-26 21:37 by Ruturaj4
[ROCm] Change run_multi_gpu set opts open-upstream
#156 opened 2024-11-26 17:28 by Ruturaj4
Terms Of Use Privacy Policy Refund Policy Impressum