lightning
torch
torchvision
torchaudio

[benchmark]
numpy
matplotlib
pandas
ucimlrepo
timm
transformers
scikit-learn

[dev]
black
black[jupyter]
ruff
isort
mypy
pytest
pytest-cov
mkdocs
mkdocs-material
mkdocstrings
mkdocstrings[python]
build
twine
