Skip to content

Pull requests: ROCm/jax

New pull request New
68 Open 715 Closed
68 Open 715 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] Add blocking Bazel PR gate
#806 opened Jun 19, 2026 by mminutoli Loading…
Cap xdist workers per GPU to avoid oversubscription
#805 opened Jun 18, 2026 by psanal35 Loading…
Isolate MIOpen cache per xdist worker to avoid lock contention
#804 opened Jun 18, 2026 by psanal35 Loading…
[ROCm] Overhaul and consolidate ROCm install documentation
#801 opened Jun 15, 2026 by JehandadKhan Collaborator Loading…
7
[ROCm] Deselect crashing FusedAttention bwd8 + add pytest-timeout (ROCM-25626)
#798 opened Jun 9, 2026 by srinivamd Loading…
4 tasks
[ROCm] Skip 6 Pallas FusedAttentionTest variants exceeding gfx942 LDS limit
#797 opened Jun 9, 2026 by srinivamd Loading…
3 tasks
4
[ROCm] xdist: select GPU from runner-allocated ROCR slice
#793 opened Jun 5, 2026 by mminutoli Loading…
1
ci: deselect PGLE/FDO profiling tests on ROCm (ROCM-25581) [v0.9.2]
#790 opened Jun 3, 2026 by srinivamd Loading…
1
Disable the cron trigger for the cont. wheel tests workflow
#789 opened Jun 3, 2026 by psanal35 Loading…
ci: deselect PGLE/FDO profiling tests on ROCm (ROCM-25581)
#788 opened Jun 3, 2026 by srinivamd Loading…
Cherry-pick ROCm fixes from PR #740 to rocm-jaxlib-v0.8.2
#786 opened Jun 2, 2026 by srinivamd Loading…
1
Cherry-pick ROCm fixes from PR #740 to rocm-jaxlib-v0.9.0
#785 opened Jun 2, 2026 by srinivamd Loading…
2
[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…
11
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
ProTip! Exclude everything labeled bug with -label:bug.