diffrax
distreqx>=0.0.2
equinox
jax
jaxtyping
optax
rich
tensorboard
tensorboardX

[compatibility]
gymnasium
stable-baselines3
gymnax
flax

[docs]
zensical
mkdocstrings-python

[mujoco]
mujoco-mjx

[render]
pygame
imageio
imageio-ffmpeg
