Skip to content

Pull requests: ROCm/jax

New pull request New
71 Open 782 Closed
71 Open 782 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

Test PR CI trigger
#877 opened Sep 1, 2026 by charleshofer Collaborator Loading…
Backport: Mount /home/runner/_work in ROCm workflow containers
#841 opened Aug 13, 2026 by mkunredd Loading…
2 tasks
Mount /home/runner/_work in ROCm workflow containers
#840 opened Aug 12, 2026 by mkunredd Loading…
2 tasks
4
Prevent job-container teardown hangs in ROCm CI
#820 opened Jul 27, 2026 by magaonka-amd Loading…
[ROCm] Pin pyenv clone to a fixed commit in Dockerfile.ms
#815 opened Jul 15, 2026 by magaonka-amd Loading…
[ROCm] Add blocking Bazel PR gate
#806 opened Jun 19, 2026 by mminutoli Loading…
1
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] 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…
ProTip! Follow long discussions with comments:>50.