einops
lightning
matplotlib
numpy
Pillow
pyyaml
scikit-image
torch
wandb
x-transformers

[testing]
tox
pytest
pytest-cov
pytest-mock
