Skip to content

Pull requests: ROCm/jax

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
Assignee
Filter by who’s assigned
Sort

Pull requests list

CI: 02/07/25 upstream sync
#225 opened Feb 7, 2025 by github-actions bot Loading…
Fix invalid lowerings for ROCm in Pallas
#223 opened Feb 6, 2025 by mrodden Loading…
CI: 02/04/25 upstream sync
#221 opened Feb 4, 2025 by github-actions bot Loading…
Use HIPBLAS_V2
#220 opened Feb 3, 2025 by hsharsha Loading…
Use bazel for PR tests
#216 opened Jan 28, 2025 by charleshofer 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…
Add gfx12xx architecture support open-upstream Tag when you want a copy of this PR to be opened on upstream
#203 opened Jan 14, 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…
[ROCm] Remove exit on error
#166 opened Nov 27, 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…
[ROCm] BUILD file fixes and linter run open-upstream Tag when you want a copy of this PR to be opened on upstream
#155 opened Nov 26, 2024 by Ruturaj4 Loading…
[ROCm] fix jax and wheelhouse relative paths open-upstream Tag when you want a copy of this PR to be opened on upstream
#153 opened Nov 26, 2024 by Ruturaj4 Loading…
ProTip! Adding no:label will show everything without a label.