compatibility-matrix
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
