black
flake8
pylint
typing-extensions
setuptools_scm
pytest
pillow
jax
flax
rlax
tyro
wandb
