Skip to content

Pull requests: ROCm/jax

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
Assignee
Filter by who’s assigned
Assigned to nobody Loading
Sort

Pull requests list

Release Prep JAX 0.9.1
#735 opened Mar 6, 2026 by gulsumgudukbay Loading…
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…
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…
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…
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…
Support wheels as build deps
#638 opened Jan 22, 2026 by alekstheod Loading…
Use xla from rocm-jaxlib-v0.8.0 release branch
#599 opened Jan 14, 2026 by alekstheod Loading…
ProTip! Adding no:label will show everything without a label.