Search papers, labs, and topics across Lattice.
MuJoCoUni is introduced as a downstream MuJoCo distribution designed for high-throughput parallel execution in online robot learning and batched physics evaluation. It provides runtime primitives for stateful environment execution, including final-state-only short stepping, sparse reset, and batched sensor evaluation. The core component, BatchEnvPool, manages per-environment mjModel copies and mjData workers within a thread pool, while maintaining upstream MuJoCo semantics for core functionalities.
Unleash parallel robot learning with MuJoCoUni, enabling high-throughput batched physics evaluation without sacrificing upstream MuJoCo semantics.
We present MuJoCoUni, a downstream MuJoCo distribution for online robot learning and batched physics evaluation. Alongside the open-loop batched trajectory generation already provided by upstream mujoco.rollout, MuJoCoUni supplies runtime primitives for stateful environment execution. The target workloads need high-throughput parallel execution while retaining upstream CPU MuJoCo semantics for models, sensors, contact, and constraints. Its core object, BatchEnvPool, is a C++/pybind11 executor that owns per-environment mjModel copies, per-thread mjData workers, and an internal thread pool. It provides final-state-only short stepping, sparse reset, reset-lifecycle domain randomization, batched sensor forward evaluation without advancing dynamics, and batched Jacobian and height-field queries. The implementation is confined to the Python binding layer; MuJoCo's solver, contact model, integrator, and core source tree retain upstream semantics. This report describes the BatchEnvPool API, implementation boundary, relationship to rollout, and the validation and benchmark scripts shipped with the open-source mujoco-uni package, which is installed with \texttt{pip install mujoco-uni}.
Tsinghua AI