torch>=2.0.0
numpy>=1.24.0
tqdm>=4.65.0

[all]
sparse-hyper-connections[dev,docs,training]

[dev]
pytest>=7.0.0
pytest-cov>=4.0.0
pytest-benchmark>=4.0.0
black>=23.0.0
isort>=5.12.0
mypy>=1.0.0
pre-commit>=3.0.0

[docs]
sphinx>=7.0.0
sphinx-rtd-theme>=1.3.0
myst-parser>=2.0.0
sphinx-autodoc-typehints>=1.24.0

[training]
tensorboard>=2.12.0
deepspeed>=0.9.0
hydra-core>=1.3.0
wandb>=0.15.0
