diff --git a/mjx/tutorial.ipynb b/mjx/tutorial.ipynb index 751a150a..905e2ae3 100644 --- a/mjx/tutorial.ipynb +++ b/mjx/tutorial.ipynb @@ -321,7 +321,7 @@ "source": [ "Now let's run the same exact simulation on the GPU device using MJX!\n", "\n", - "In the example below, we use `mjx.step` instead of `mujoco.mj_step`, and we also [`jax.jit`](https://jax.readthedocs.io/en/latest/jax-101/02-jitting.html) the `mjx.step` so that it runs efficiently on the GPU. After each step, we convert the `mjx.Data` back to `mjData` so that we can use the MuJoCo renderer.\n" + "In the example below, we use `mjx.step` instead of `mujoco.mj_step`, and we also [`jax.jit`](https://jax.readthedocs.io/en/latest/jax-101/02-jitting.html) the `mjx.step` so that it runs efficiently on the GPU. For each frame, we convert the `mjx.Data` back to `mjData` so that we can use the MuJoCo renderer.\n" ] }, { @@ -563,7 +563,7 @@ "\n", "Let's instantiate the environment and visualize a short rollout.\n", "\n", - "NOTE: Since episodes terminates early if the torso is below the healthy z-range, the only relevant contacts for this task are between the feet and the plane. We turn off other contacts." + "NOTE: Since episodes terminate early if the torso is below the healthy z-range, the only relevant contacts for this task are between the feet and the plane. We turn off other contacts." ] }, {