Oven

jax-rocm60-plugin0.5.0

Published a month ago

JAX Plugin for AMD GPUs (ROCm:unknown)

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.5.0