datasets
huggingface_hub
fpdf
pandas
numpy
matplotlib
scikit-learn
pyarrow
seaborn

[dev]
pytest>=3.9
pytest-cov
black
twine
setuptools
wheel
pylint
tensorflow<2.16,>=2.13
torch
torchvision

[lightning]
lightning

[pytorch]
torch
torchvision

[tensorflow]
tensorflow<2.16,>=2.13

[tf]
tensorflow<2.16,>=2.13

[torch]
torch
torchvision

[wandb]
wandb>=0.15
