forked from jax-ml/jax
-
Notifications
You must be signed in to change notification settings - Fork 10
Pull requests: ROCm/jax
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
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…
[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
[ROCm] xdist: select GPU from runner-allocated ROCR slice
#793
opened Jun 5, 2026 by
mminutoli
Loading…
ci: deselect PGLE/FDO profiling tests on ROCm (ROCM-25581) [v0.9.2]
#790
opened Jun 3, 2026 by
srinivamd
Loading…
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…
Cherry-pick ROCm fixes from PR #740 to rocm-jaxlib-v0.9.0
#785
opened Jun 2, 2026 by
srinivamd
Loading…
[ROCm] Skip TF32 dot algorithm tests on Navi/RDNA GPUs
#783
opened Jun 1, 2026 by
srinivamd
Loading…
3 tasks
Bundle specific and generic arch build targets together.
#780
opened May 20, 2026 by
tsrw2048
Loading…
Fix env var leak in xla_bridge_test register_plugin tests
#775
opened May 14, 2026 by
tsrw2048
Loading…
Skip ann_test test_pmap on ROCm due to IndivisibleError
#773
opened May 14, 2026 by
AratiGanesh
Loading…
[ROCm] Skip core_test reduction autotuner failures on Navi/RDNA GPUs (gfx11xx)
#758
opened Apr 28, 2026 by
CSkmd
Loading…
[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…
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…
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…
Previous Next
ProTip!
Add no:assignee to see everything that’s not assigned.