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] 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…
[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…
Previous Next
ProTip!
no:milestone will show everything without a milestone.