Skip to content

Pull requests: ROCm/rocm-jax

New pull request New
43 Open 279 Closed
43 Open 279 Closed
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
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Add AWS CLI to manylinux image
#346 opened Mar 9, 2026 by charleshofer Loading…
Fixed dev build v0.9.0
#345 opened Mar 9, 2026 by zahiqbal Loading…
1 task
enable rocprof sdk
#344 opened Mar 6, 2026 by Ruturaj4 Loading…
1 task done
Add --no-cache flag to build_dockers and build_base_dockers
#343 opened Mar 6, 2026 by mminutoli Loading…
1 task done
6
Prepare for JAX 0.9.1 release
#338 opened Mar 3, 2026 by gulsumgudukbay Loading…
1 task done
Update TE to new build
#337 opened Mar 3, 2026 by gabeweisz • Draft
Update README to pull latest nightly image
#331 opened Mar 2, 2026 by charleshofer Loading…
Rework plugin/PJRT wheel build to use Bazel Build
#328 opened Mar 2, 2026 by lucbruni-amd Loading…
1 task done
2
Enable rbe for multigpu tests
#322 opened Feb 19, 2026 by alekstheod Loading…
Introduce tsan pipeline
#321 opened Feb 19, 2026 by alekstheod Loading…
V0.9.0 merger
#320 opened Feb 18, 2026 by Ruturaj4 Loading…
1 task done
Add dynamic git repository, make it possible to override jax/xla deps…
#315 opened Feb 18, 2026 by alekstheod Loading…
Bump xla
#309 opened Feb 12, 2026 by alekstheod Loading…
3
[WIP] Make the nightly job use TheRock instead of ROCm
#305 opened Feb 11, 2026 by charleshofer • Draft
5 tasks
3
Refactor build wheels pipeline to build wheels and run pytest
#301 opened Feb 10, 2026 by alekstheod Loading…
10
Bump up jax and xla versions
#300 opened Feb 10, 2026 by Ruturaj4 Loading…
1 task done
2
[WIP] Remove patchelf usage
#299 opened Feb 10, 2026 by alekstheod Loading…
17
Rocm jaxlib v0.8.0 fixdevtar
#292 opened Feb 5, 2026 by kiran-thumma Loading…
1 task
3
[WIP] Migrate jax dep to bzlmod
#282 opened Jan 26, 2026 by alekstheod Loading…
Update model performance workflows with ROCm 7.2.0 public release
#279 opened Jan 23, 2026 by psanal35 Loading…
1
add hardcoded or fixed number to fetch rocm 7.2.0
#270 opened Jan 22, 2026 by Ruturaj4 Loading…
1 task done
4
Remove --rbe from the build_wheel.yml
#264 opened Jan 19, 2026 by Ruturaj4 Loading…
1 task done
3
Run single and multi GPU tests in parallel and upload combined results
#257 opened Jan 15, 2026 by psanal35 Loading…
2
[WIP] Stack direct build without Makefile
#246 opened Jan 9, 2026 by i-chaochen Loading…
12
To build and install jax, jaxlib, pjrt and plugin at once
#244 opened Jan 9, 2026 by i-chaochen Loading…
27
ProTip! What’s not been updated in a month: updated:<2026-02-09.