temporaldata>=0.1.3
numpy
torch~=2.0
einops~=0.6.0
hydra-core~=1.3.2
torchtyping~=0.1
torchmetrics>=1.6.0
pydantic~=2.0
rich

[dev]
pytest
black==24.2.0
pre-commit>=3.5.0
flake8
