Skip to content

Pull requests: ROCm/rocm-jax

New pull request New
39 Open 262 Closed
39 Open 262 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

Fix: Add vim-common package to manylinux Docker image for XLA build
#324 opened Feb 20, 2026 by Ruturaj4 Loading…
1 task done
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
Install Bazel into Ubuntu base images
#319 opened Feb 18, 2026 by charleshofer • Draft
1 task
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
Pytest JSON results to DB
#294 opened Feb 5, 2026 by psanal35 Loading…
Rocm jaxlib v0.8.0 fixdevtar
#292 opened Feb 5, 2026 by kiran-thumma Loading…
1 task
3
make env var paths independent of ROCm version
#285 opened Jan 28, 2026 by JehandadKhan Loading…
1 task done
1
[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
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
ProTip! Mix and match filters to narrow down what you’re looking for.