Skip to content

Pull requests: ROCm/jax

New pull request New
36 Open 661 Closed
36 Open 661 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Implement Mosaic GPU detection and Auto-Skips open-upstream Tag when you want a copy of this PR to be opened on upstream
#714 opened Feb 21, 2026 by gulsumgudukbay Loading…
fix poor performance of slogdet when input dimension n=1 or 2 cherry-pick-candidate Mark a PR to be cherry-picked into the next ROCm JAX. Remove IIF the latest upstream contain the PR. enhancement New feature or request
#710 opened Feb 10, 2026 by cj401-amd Loading…
1
Skip test_tridiagonal_solve_grad test 0.8.0 cherry-pick-candidate Mark a PR to be cherry-picked into the next ROCm JAX. Remove IIF the latest upstream contain the PR. open-upstream Tag when you want a copy of this PR to be opened on upstream
#701 opened Feb 9, 2026 by AratiGanesh Loading…
Skip testEighTinyNorm test 0.9.0 cherry-pick-candidate Mark a PR to be cherry-picked into the next ROCm JAX. Remove IIF the latest upstream contain the PR. open-upstream Tag when you want a copy of this PR to be opened on upstream
#699 opened Feb 9, 2026 by AratiGanesh Loading…
2
Enable scaled matmul tests on ROCm
#678 opened Jan 30, 2026 by magaonka-amd • Draft
Rocm jaxlib v0.9.0
#671 opened Jan 28, 2026 by Ruturaj4 Loading…
1 task done
Test pytest abort plugin
#670 opened Jan 28, 2026 by gulsumgudukbay Loading…
1 task done
update with ROCm profiling tests with kernel_details
#664 opened Jan 28, 2026 by cj401-amd Loading…
remove skipTest associated to hipsparse transpose
#661 opened Jan 27, 2026 by kliegeois Loading…
2
Support wheels as build deps
#638 opened Jan 22, 2026 by alekstheod Loading…
Add script for running bazel tests on ROCm
#632 opened Jan 21, 2026 by charleshofer • Draft
1
Use xla from rocm-jaxlib-v0.8.0 release branch
#599 opened Jan 14, 2026 by alekstheod Loading…
3
Add CI workflow for Pytest ROCm
#594 opened Jan 12, 2026 by mminutoli Loading…
1 task done
Upstream ci download jax rocm wheels
#592 opened Jan 10, 2026 by mminutoli • Draft
1 task
Adding run_pytest_rocm.sh
#591 opened Jan 9, 2026 by gulsumgudukbay Loading…
1 task done
update a test for checking zero ROCm GPU event cherry-pick-candidate Mark a PR to be cherry-picked into the next ROCm JAX. Remove IIF the latest upstream contain the PR. open-upstream Tag when you want a copy of this PR to be opened on upstream
#585 opened Dec 18, 2025 by cj401-amd Loading…
13
Use plain bazel to test jax, use hermetic rocm dependency
#584 opened Dec 18, 2025 by alekstheod Loading…
7
Fix/pallas tests shared memory
#581 opened Dec 15, 2025 by AratiGanesh Loading…
1 task
4
Create rocm-test-requirements.txt
#579 opened Dec 12, 2025 by jvadakat Loading…
1 task
@JehandadKhan @jvadakat
[0.6.0-UT] Fix Test Shard
#519 opened Jul 29, 2025 by AratiGanesh Loading…
2
[0.6.0-UT] Adjust tolerance for reciprocal in unary primitives test
#518 opened Jul 25, 2025 by AratiGanesh Loading…
2
CI: 06/20/25 upstream sync
#477 opened Jun 20, 2025 by rocm-repo-management-api-2 bot Loading…
Add rocm perf yml file
#418 opened May 12, 2025 by Ruturaj4 Loading…
14
Add linear algebra tests to our CI workflow
#388 opened Apr 28, 2025 by charleshofer Loading…
gfx950 platform support
#381 opened Apr 23, 2025 by zahiqbal Loading…
4
ProTip! Type g i on any issue or pull request to go back to the issue listing page.