numpy<2.0,>=1.22
termcolor>=1.1.0
scikit-learn
pandas>=1.4.3
tabulate>=0.8.10
torch>=1.12.0
torchvision>=0.13.0
pytorch-lightning>=1.6.3
tensorboard>=2.9.1
nbclient>=0.10.1
seaborn>=0.13.2
scienceplots>=2.1.1

[benchmarks]
tqdm>=4.64.0
torch-scatter>=2.0.9
torchmetrics>=0.9.2

[complete]
torch-choice[benchmarks,dev,docs,notebooks]
absl-py>=1.2.0
aiohttp>=3.8.1
tensorboard>=2.9.1
PyYAML>=6.0
requests>=2.28.1
scipy>=1.8.1

[dev]
pytest>=6.0
pytest-cov
black
isort
flake8
mypy
pre-commit

[docs]
mkdocs>=1.4.0
mkdocs-material
mkdocs-jupyter
matplotlib
seaborn

[notebooks]
jupyter>=1.0.0
ipykernel
matplotlib>=3.5.0
seaborn>=0.11.0
plotly>=5.0.0
