Skip to content

Pull requests: ROCm/jax

New pull request New
57 Open 704 Closed
57 Open 704 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

[ROCm] Skip TF32 dot algorithm tests on Navi/RDNA GPUs
#783 opened Jun 1, 2026 by srinivamd Loading…
3 tasks
Add docker image to use hermetic llvm and sysroot
#782 opened May 29, 2026 by hsharsha Loading…
9
[ROCm] Bake TheRock-aware RUNPATH into rocm plugin/pjrt wheels
#781 opened May 28, 2026 by gulsumgudukbay Loading…
1
Bundle specific and generic arch build targets together.
#780 opened May 20, 2026 by tsrw2048 Loading…
7
Fix env var leak in xla_bridge_test register_plugin tests
#775 opened May 14, 2026 by tsrw2048 Loading…
1
Skip ann_test test_pmap on ROCm due to IndivisibleError
#773 opened May 14, 2026 by AratiGanesh Loading…
jaxlib/gpu/rnn_kernels.cc: remove premature input_tensor_desc destruction (ROCM-21859)
#765 opened Apr 30, 2026 by srinivamd • Draft
[ROCm] Skip core_test reduction autotuner failures on Navi/RDNA GPUs (gfx11xx)
#758 opened Apr 28, 2026 by CSkmd Loading…
4
[ROCm]---Theory--- force ROCm xdist worker GPU env override
#749 opened Apr 10, 2026 by magaonka-amd Loading…
port rocm/jax 0.9.2 cherrypicks to amd-main branch (#740)
#741 opened Mar 23, 2026 by magaonka-amd Loading…
Fix: Scaled Matmul at rocm 9.0
#739 opened Mar 17, 2026 by shurale-nkn Loading…
Fix: Scaled Matmul at rocm 9.1
#738 opened Mar 17, 2026 by shurale-nkn Loading…
Add ROCm gesdd SVD backend, export back-compat data, and relaxed GPU … enhancement New feature or request
#734 opened Mar 6, 2026 by cj401-amd Loading…
1
Add ROCm gesdd SVD backend, export back-compat data, and relaxed GPU … enhancement New feature or request
#733 opened Mar 6, 2026 by cj401-amd Loading…
@cj401-amd
1
update test
#732 opened Mar 5, 2026 by kiran-thumma • Draft
update test
#731 opened Mar 5, 2026 by kiran-thumma • Draft
Remove input_tensor_desc destruction in RNN kernels cherry-pick-candidate Mark a PR to be cherry-picked into the next ROCm JAX. Remove IIF the latest upstream contain the PR.
#729 opened Mar 5, 2026 by magaonka-amd Loading…
Fix HIP memory leaks in RNN kernels
#728 opened Mar 5, 2026 by magaonka-amd • Draft
Fix HIP memory leaks in RNN kernels
#727 opened Mar 5, 2026 by magaonka-amd • Draft
3
Skip test_pmap on ROCm due to IndivisibleError with new pmap SPMD tiling
#725 opened Mar 4, 2026 by AratiGanesh Loading…
[Testing do not merge] Add ROCm wheel build and test pipeline to continuous CI
#723 opened Mar 4, 2026 by alekstheod • Draft
Fix batched Cholesky OOM on ROCm by bypassing hipSolver's external hipMalloc
#717 opened Feb 26, 2026 by FlemingH Loading…
1
2
Improve ROCm pytest results handling
#716 opened Feb 25, 2026 by psanal35 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…
ProTip! Follow long discussions with comments:>50.