Skip to content

Pull requests: ROCm/jax

New pull request New
29 Open 541 Closed
29 Open 541 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 KeyError for bytes_reservable_limit on ROCm
#588 opened Jan 6, 2026 by magaonka-amd Loading…
Mark ROCm support for gtsv2 as stable
#587 opened Jan 6, 2026 by tsrw2048 Loading…
update a test for checking zero ROCm GPU event 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
#585 opened Dec 18, 2025 by cj401-amd Loading…
8
Use plain bazel to test jax, use hermetic rocm dependency
#584 opened Dec 18, 2025 by alekstheod Loading…
7
Fix/pallas tests shared memory
#581 opened Dec 15, 2025 by AratiGanesh Loading…
1 task
4
GESVDJ support for ROCm GPUs in JAX
#580 opened Dec 12, 2025 by tsrw2048 Loading…
1
Create rocm-test-requirements.txt
#579 opened Dec 12, 2025 by jvadakat Loading…
1 task
@JehandadKhan @jvadakat
[0.6.0-UT] unskip selected test cases
#534 opened Sep 9, 2025 by phambinhfin Loading…
6
Fix GPU handle pool singleton aliasing with tag-based template separa…
#524 opened Aug 1, 2025 by phambinhfin Loading…
@Arech8
14
[0.6.0-UT] Fix Test Shard
#519 opened Jul 29, 2025 by AratiGanesh Loading…
2
[0.6.0-UT] Adjust tolerance for reciprocal in unary primitives test
#518 opened Jul 25, 2025 by AratiGanesh Loading…
2
add rocprofiler_configure as a global
#479 opened Jun 20, 2025 by cj401-amd Loading…
5
add rocprofiler_configure as a global
#478 opened Jun 20, 2025 by cj401-amd Loading…
CI: 06/20/25 upstream sync
#477 opened Jun 20, 2025 by rocm-repo-management-api-2 bot Loading…
Add HIP runtime dependency for hybird library
#461 opened Jun 6, 2025 by phambinhfin Loading…
1
Add rocm perf yml file
#418 opened May 12, 2025 by Ruturaj4 Loading…
14
Add linear algebra tests to our CI workflow
#388 opened Apr 28, 2025 by charleshofer Loading…
gfx950 platform support
#381 opened Apr 23, 2025 by zahiqbal Loading…
4
Fix CI cleanup race condition
#349 opened Apr 8, 2025 by charleshofer Loading…
1
Update latest xla hash for JAX branch 0.5.0
#337 opened Apr 2, 2025 by Ruturaj4 Loading…
2
Find pyversion at build time 0.5.0
#328 opened Mar 28, 2025 by mrodden Loading…
[rocm-main]: Canonicalize python build versions
#321 opened Mar 26, 2025 by mrodden Loading…
18
Add support for ROCm wheel based install
#305 opened Mar 21, 2025 by mrodden Loading…
Update XLA Commit hash
#266 opened Mar 7, 2025 by JehandadKhan Loading…
[ROCm] Implement RNN support open-upstream Tag when you want a copy of this PR to be opened on upstream
#171 opened Dec 2, 2024 by Ruturaj4 Loading…
2
ProTip! Updated in the last three days: updated:>2026-01-04.