Simple-RJX (Demonstration Only) 🚀

Simple-RJX is a streamlined collection of MJX environments designed for END-TO-END robotics training using reinforcement learning

• Built for simplicity
• Seamlessly integrates with the Brax training pipeline (a well respected JAX based PPO baseline in the community)

With GPU-accelerated parallelism, you can set up, train, and achieve results in minutes no hassle, just plug and play within minutes.

Library

Fetch

Training Times (RTX 4090 Mobile)
Mel humanoid	Mel Quadruped	Unitree H1	Unitree G1	Google Barkour Vb	Unitree Go2
60 minutes	43 minutes	TBD minutes	TBD minutes	18 minutes	24 minutes

Reproduce the environments by training locally or instantly using the ***** web links under each environment

Long Jump

Training Times (RTX 4090 Mobile)
Mel humanoid	Mel Quadruped	Unitree H1	Unitree G1	Google Barkour Vb	Unitree Go2
17 minutes	TBD minutes	TBD minutes	TBD minutes	TBD minutes	TBD minutes

Reproduce the environments by training locally or instantly using the ***** web links under each environment

Sprints

Training Times (RTX 4090 Mobile)
Mel humanoid	Mel Quadruped	Unitree H1	Unitree G1	Google Barkour Vb	Unitree Go2
5 minutes	5 minutes	7 minutes	27 minutes	11 minutes	10 minutes

Reproduce the environments by training locally or instantly using the ***** web links under each environment

Stairs

Training Times (RTX 4090 Mobile)
Mel humanoid	Mel Quadruped	Unitree H1	Unitree G1	Google Barkour Vb	Unitree Go2
120 minutes	12 minutes	14 minutes	45 minutes	28 minutes	52 minutes

Reproduce the environments by training locally or instantly using the ***** web links under each environment