torch
torchvision
torchaudio
lightning
kaggle
seaborn
pandas
diffusers[torch]
transformers
ipython
hydra-core
wandb
POT
tables
einops
SciencePlots

[test]
pytest
coverage
mypy
black
pytest-cov
pytest-mypy
pytest-black
pytest-isort
pytest-flakes
pre-commit
