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

Set higher tolerance for failing fp8 tests
#306 opened Mar 21, 2025 by Ruturaj4 Loading…
Add support for ROCm wheel based install
#305 opened Mar 21, 2025 by mrodden Loading…
CI: 03/21/25 upstream sync
#304 opened Mar 21, 2025 by rocm-repo-management-api-2 bot Loading…
Fix lowering code for ROCm RNN
#303 opened Mar 20, 2025 by Ruturaj4 Loading…
CI: 03/20/25 upstream sync
#300 opened Mar 20, 2025 by rocm-repo-management-api-2 bot Loading…
Add back raw totals in JSON reports
#281 opened Mar 14, 2025 by charleshofer Loading…
Update XLA Commit hash
#266 opened Mar 7, 2025 by JehandadKhan 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…
[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! Type g p on any issue or pull request to go back to the pull request listing page.