Skip to content

Pull requests: ROCm/rocm-jax

New pull request New
27 Open 200 Closed
27 Open 200 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

Stack direct build without Makefile
#246 opened Jan 9, 2026 by i-chaochen Loading…
Fix abort detection
#245 opened Jan 9, 2026 by gulsumgudukbay Loading…
1 task done
To build and install jax, jaxlib, pjrt and plugin at once
#244 opened Jan 9, 2026 by i-chaochen Loading…
To build and install jax, jaxlib, pjrt and plugin at once.
#243 opened Jan 8, 2026 by i-chaochen Loading…
1
Revert "Improve check for AMD GPU availability; drop the patchfile (#226)"
#241 opened Jan 7, 2026 by psanal35 Loading…
1
[WIP] Introduce jax nightly multigpu ci script
#239 opened Dec 30, 2025 by alekstheod Loading…
17
Bazelize rocm jax
#238 opened Dec 24, 2025 by alekstheod Loading…
Add docs on bulid.py
#218 opened Dec 11, 2025 by charleshofer • Draft
18
Update JAX & XLA commits in bazel files
#217 opened Dec 10, 2025 by yeandy • Draft
1 task
Use ROCm 7.2.0 RC build 14 for nightly
#214 opened Dec 9, 2025 by charleshofer Loading…
GitHub Actions workflows to support running on k8's
#210 opened Dec 5, 2025 by okakarpa Loading…
6
[POC] Pre-build manylinux and devsetup docker images
#209 opened Dec 5, 2025 by charleshofer • Draft
1 task
add the bazel cache to the github cache action
#205 opened Dec 4, 2025 by JehandadKhan • Draft
1 task
Fix symlink command from ENV to RUN
#198 opened Nov 27, 2025 by Ruturaj4 Loading…
1 task done
4
Add RBE secrets to nightly
#189 opened Nov 20, 2025 by charleshofer Loading…
1
Test new runners
#187 opened Nov 20, 2025 by JehandadKhan Loading…
1 task
Build jaxlib wheel
#180 opened Nov 18, 2025 by zahiqbal Loading…
1 task
Handle deselected/ignored tests and update skip labels in DB
#177 opened Nov 14, 2025 by psanal35 • Draft
Add symbols to wheel builds in the CI
#174 opened Nov 13, 2025 by JehandadKhan • Draft
15
update manylinux dockerfile fix
#170 opened Nov 7, 2025 by kiran-thumma • Draft
1 task
Update plugin_setup.py and pjrt setup to add JAX devs dl contact
#168 opened Nov 5, 2025 by gulsumgudukbay Loading…
1 task done
Fixed buidling docker with released-rocm or from ci build
#165 opened Nov 3, 2025 by zahiqbal Loading…
1 task
1
Use the app to log in to GHCR for nightly test runs
#154 opened Oct 23, 2025 by charleshofer • Draft
1 task
Add version check to ROCm PJRT plugin
#130 opened Sep 25, 2025 by JehandadKhan • Draft
Store JAX commit SHA in DB in nightly runs
#125 opened Sep 22, 2025 by psanal35 Loading…
ProTip! no:milestone will show everything without a milestone.