numpy
jax
flax
chex
gymnax

[examples]
distrax
optax
orbax-checkpoint==0.5.3
orbax
wandb
pillow
imageio
moviepy
