diff --git a/doc/mjwarp/index.rst b/doc/mjwarp/index.rst index e6d1bf6a..d2bc1e92 100644 --- a/doc/mjwarp/index.rst +++ b/doc/mjwarp/index.rst @@ -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 `__. Please see the +`Warp Interoperability `__ documentation for details. + +Additionally, :ref:`MJX ` provides a JAX API for a subset of MJWarp's :ref:`API `. The backend is +specified with ``impl='warp'``. + +**Does MJWarp work with PyTorch?** + +Yes. MJWarp is interoperable with `PyTorch `__. Please see the +`Warp Interoperability `__ documentation for +details. + +**How to train policies with MJWarp physics?** + +For examples that train policies with MJWarp physics, please see: + +- `Isaac Lab `__: Train via + `Newton API `__. +- `mjlab `__: Train directly with MJWarp using PyTorch. +- `MuJoCo Playground `__: Train via :ref:`MJX API `. + Orientation representation --------------------------