jax
flax
numpy
pygame
matplotlib
imageio

[dev]
ruff
pre-commit
