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