torch
tensordict
torchmetrics
einops
scipy
tqdm
numpy

[dev]
pre-commit
ruff
toml-sort

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

[tests]
torch
tensordict
torchmetrics
einops
scipy
tqdm
numpy
pytest
pytest-cov
pandas
