Skip to content

Pull requests: ROCm/jax

New pull request New
50 Open 667 Closed
50 Open 667 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

Fix: Scaled Matmul at rocm
#736 opened Mar 15, 2026 by shurale-nkn Loading…
Release Prep JAX 0.9.1
#735 opened Mar 6, 2026 by gulsumgudukbay 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
Revert "Fix HIP memory leaks in RNN kernels"
#730 opened Mar 5, 2026 by magaonka-amd Loading…
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…
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
ProTip! Mix and match filters to narrow down what you’re looking for.