Add Frequently Asked Questions: Learning frameworks to MuJoCo Warp documentation.
PiperOrigin-RevId: 836231961 Change-Id: If3d00e7dfb9ebe49a034ea8936d2ac1dd33d0979
This commit is contained in:
committed by
Copybara-Service
parent
cb391962ca
commit
2ca56179a7
@@ -347,6 +347,32 @@ It is possible to override the field shape and set the field values after graph
|
|||||||
Frequently Asked Questions
|
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
|
Orientation representation
|
||||||
--------------------------
|
--------------------------
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user