jax
distrax
optax
flax
numpy
pygame
gymnax
chex
matplotlib
imageio

[dev]
black
pre-commit
