Fix: shim jax.device_put_replicated removed in JAX 0.10

G1 rough-terrain training app (MuJoCo Playground + Brax PPO) (#1)

initial commit