google / brax

Massively parallel rigidbody physics simulation on accelerator hardware.
Apache License 2.0
2.14k stars 234 forks source link

[MJX] Model loading errors in `brax.io`, for features supported by MJX #459

Closed varadVaidya closed 4 months ago

varadVaidya commented 4 months ago

With the MuJoCo 3.1.2 update, the site transmission was added to MJX, and a new pipeline to train RL agents using PipelineEnv was added brax. However while loading the model, through this pipeline, errors regarding RK4 integration scheme, and site transmission, are thrown, even though the pipeline is set to mjx.

varadVaidya commented 4 months ago

Solved in #1422