Add Frequently Asked Questions: Learning frameworks to MuJoCo Warp documentation.

PiperOrigin-RevId: 836231961
Change-Id: If3d00e7dfb9ebe49a034ea8936d2ac1dd33d0979
This commit is contained in:
Taylor Howell
2025-11-24 08:13:48 -08:00
committed by Copybara-Service
parent cb391962ca
commit 2ca56179a7
+26
View File
@@ -347,6 +347,32 @@ It is possible to override the field shape and set the field values after graph
Frequently Asked Questions
==========================
Learning frameworks
-------------------
**Does MJWarp work with JAX?**
Yes. MJWarp is interoperable with `JAX <https://jax.readthedocs.io/>`__. Please see the
`Warp Interoperability <https://nvidia.github.io/warp/modules/interoperability.html#jax>`__ documentation for details.
Additionally, :ref:`MJX <mjx>` provides a JAX API for a subset of MJWarp's :ref:`API <api.rst>`. The backend is
specified with ``impl='warp'``.
**Does MJWarp work with PyTorch?**
Yes. MJWarp is interoperable with `PyTorch <https://pytorch.org>`__. Please see the
`Warp Interoperability <https://nvidia.github.io/warp/modules/interoperability.html#pytorch>`__ documentation for
details.
**How to train policies with MJWarp physics?**
For examples that train policies with MJWarp physics, please see:
- `Isaac Lab <https://github.com/isaac-sim/IsaacLab/tree/feature/newton>`__: Train via
`Newton API <https://github.com/newton-physics/newton>`__.
- `mjlab <https://github.com/mujocolab/mjlab>`__: Train directly with MJWarp using PyTorch.
- `MuJoCo Playground <https://github.com/google-deepmind/mujoco_playground>`__: Train via :ref:`MJX API <mjx>`.
Orientation representation
--------------------------