MJX RL getting started tutorial scripts with high batch rendering with Warp?

[jrmch12]

As madrona mjx is deprecated, are there some tutorials for MJX Brax RL with Warp?

[btaba]

Pushing out some changes for vision training which you'll be able to run via:

uv run --default-index https://pypi.org/simple python learning/train_jax_ppo.py --env_name=PandaPickCubeCartesian --vision --impl=warp

[btaba]

uv --no-config run train-jax-ppo --env_name CartpoleBalance --impl warp --vision on d43c721 should do it