torch
tensordict
torchmetrics
einops
scipy
tqdm
numpy

[docs]
mkdocs
mkdocs-material
mknotebooks
mkdocstrings
mkdocstrings-python
mkdocs-exclude

[tests]
torchmil
pytest
pytest-cov
pandas
