
[:python_version < "3.10"]
typing_extensions>=4.0

[dev]
pytest>=6.0
pytest-cov
pytest-xdist
h5py
numpy
pillow
ruff
mypy
torch>=2.5.1
torchvision
tensorboard

[docs]
sphinx>=5.0
sphinx-rtd-theme>=1.0
myst-parser>=0.18

[dynamo]
torch>=2.5.1
