Oven

jax-rocm60-plugin0.4.35

Published 2 months ago

JAX Plugin for AMD GPUs

pip install jax-rocm60-plugin

Package Downloads

Weekly Downloads

Monthly Downloads

Authors

[email protected]

Project URLs

Homepage

Requires Python

>=3.9

Dependencies

jax-rocm60-pjrt==0.4.35