OpenPI (JAX / Orbax) checkpoint. Normalization statistics are in assets/axis-v0.1-task501-runtime-v1/norm_stats.json.
assets/axis-v0.1-task501-runtime-v1/norm_stats.json
-