forked from jax-ml/jax
-
Notifications
You must be signed in to change notification settings - Fork 4
Pull requests: ROCm/jax
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
[0.6.0-UT] Detect aborted tests and add abort info to UT logs
#526
opened Aug 3, 2025 by
gulsumgudukbay
Loading…
Fix GPU handle pool singleton aliasing with tag-based template separa…
#524
opened Aug 1, 2025 by
phambinhfin
Loading…
[0.6.0-UT] Adjust tolerance for reciprocal in unary primitives test
#518
opened Jul 25, 2025 by
AratiGanesh
Loading…
Fix lowering code for ROCm RNN
open-upstream
Tag when you want a copy of this PR to be opened on upstream
#303
opened Mar 20, 2025 by
Ruturaj4
Loading…
Ci rocm setup fix
open-upstream
Tag when you want a copy of this PR to be opened on upstream
#238
opened Feb 25, 2025 by
Ruturaj4
Loading…
Fix run_multi_gpu script multi-gpu issue and refactor code
open-upstream
Tag when you want a copy of this PR to be opened on upstream
#206
opened Jan 15, 2025 by
Ruturaj4
Loading…
Fix hybrid type for ROCm
open-upstream
Tag when you want a copy of this PR to be opened on upstream
#192
opened Jan 6, 2025 by
Ruturaj4
Loading…
Create PyPI wheel upload script and setup.py bindings
open-upstream
Tag when you want a copy of this PR to be opened on upstream
#174
opened Dec 6, 2024 by
Ruturaj4
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…
Update http to https in amd artifactory url.
open-upstream
Tag when you want a copy of this PR to be opened on upstream
#164
opened Nov 26, 2024 by
Ruturaj4
Loading…
Previous Next
ProTip!
Type g p on any issue or pull request to go back to the pull request listing page.