Add documentation for MJX API

PiperOrigin-RevId: 860108133 Change-Id: I85469efcfe49d320ef8b89c9d174f33499a22b31
2026-01-23 07:52:27 -08:00
parent c55c9a7bf1
commit 4a64017a5f
7 changed files with 204 additions and 24 deletions
@@ -4,6 +4,11 @@
 MuJoCo XLA (MJX)
 ================

+.. toctree::
+    :hidden:
+
+    API <mjx_api.rst>
+
 Starting with version 3.0.0, MuJoCo includes MuJoCo XLA (MJX) under the
 `mjx <https://github.com/google-deepmind/mujoco/tree/main/mjx>`__ directory.  MJX allows MuJoCo to run on compute
 hardware supported by the `XLA <https://www.tensorflow.org/xla>`__ compiler via the