Package search
jax-rocm60-plugin
- Summary: JAX Plugin for AMD GPUs (ROCm:unknown)
- Author: Ruturaj4
- Homepage: https://github.com/jax-ml/jax
- Source: https://github.com/jax-ml/jax
- Repo profile
- Number of releases: 2
- First release: 0.4.35 on 2024-12-06T23:41:59
- Latest release: 0.5.0 on 2025-03-20T14:20:13
| Dependency | Optional |
|---|---|
| jax-rocm60-pjrt | false |