forked from jax-ml/jax
-
Notifications
You must be signed in to change notification settings - Fork 5
Pull requests: ROCm/jax
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
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…
[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…
[ROCm] Add argument to override base docker in dockerfile
open-upstream
Tag when you want a copy of this PR to be opened on upstream
#163
opened Nov 26, 2024 by
Ruturaj4
Loading…
[ROCm] ci build and dockerfile changes
open-upstream
Tag when you want a copy of this PR to be opened on upstream
#161
opened Nov 26, 2024 by
Ruturaj4
Loading…
[ROCm] Change run_multi_gpu set opts
open-upstream
Tag when you want a copy of this PR to be opened on upstream
#156
opened Nov 26, 2024 by
Ruturaj4
Loading…
ProTip!
no:milestone will show everything without a milestone.