Run JAX and MuJoCo MJX on AMD Ryzen AI MAX (Strix Halo, gfx1151) from pip wheels - no system ROCm - with a hang-safe check that tells you which wheel set works
What is the YauhenBichel/strix-halo-jax GitHub project? Description: "Run JAX and MuJoCo MJX on AMD Ryzen AI MAX (Strix Halo, gfx1151) from pip wheels - no system ROCm - with a hang-safe check that tells you which wheel set works". Written in Python. Explain what it does, its main use cases, key features, and who would benefit from using it.
Question is copied to clipboard — paste it after the AI opens.
Clone via HTTPS
Clone via SSH
Download ZIP
Download main.zipReport bugs or request features on the strix-halo-jax issue tracker:
Open GitHub Issues