PyPI page
Home page
Author:
Yauhen Bichel
Summary:
Run JAX and MuJoCo MJX on the AMD Ryzen AI MAX (Strix Halo, gfx1151) GPU from pip wheels, and jaxcheck: a hang-safe test of which wheel set works
Latest version:
0.1.1
Optional dependencies:
jax
|
jax-rocm7-pjrt
|
jax-rocm7-plugin
|
jaxlib
|
mujoco
|
mujoco-mjx
|
playground
|
rocm
|
rocm-sdk-core
|
rocm-sdk-libraries-gfx1151
Downloads last day:
4
Downloads last week:
11
Downloads last month:
222
7-day smoothing