Package profile
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-06
- Latest release: 0.5.0 on 2025-03-20
Releases
PyPI Downloads
Dependencies
Jax-rocm60-plugin has one dependency (non-optional).| Dependency | Optional |
|---|---|
| jax-rocm60-pjrt | false |