jax
flax
chex
gymnax
tyro
distrax
tensorboard
tensorboardX
wandb
pyyaml

[utils]
optax
numpy
tqdm
pandas
orbax-checkpoint
pytest

[viz]
scipy
imageio
pygame
matplotlib
pillow
seaborn
ipython
ipywidgets
