torch>=2.0
pydantic

[distributed]
accelerate>=0.27

[dynamics]
torchdiffeq>=0.2.3

[test]
pytest
safetensors
scipy
