[Feature]: JAX 0.8.0 is supported from rocm 7.2. Need to update rocAL
Open
@rrawther is already working on this.
Since Mar 16, 2026.
enhancement
- Dominant language
- C++
- Stars
- 25
- Forks
- 26
- Avg merge
- 5d 1h
- Merged PRs (30d)
- 2
Description
Suggestion Description
Update rocAL JAX test and Jupyter notebook to support 0.8.0.
PositionalSharding is depracated. Need to update to use NamedSharding.
https://rocm.docs.amd.com/en/latest/compatibility/compatibility-matrix.html
https://docs.jax.dev/en/latest/jax.sharding.html
Operating System
No response
GPU
No response
ROCm Component
No response
Contributor guide
No contributing guide indexed for this repository
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Assessment
This issue has not been assessed yet.