torch>=2.0
numpy>=1.24
tqdm>=4.65

[all]
alphadynamics[dev,paper,viz]

[dev]
pytest>=7.4
pytest-cov>=4.1
ruff>=0.4

[paper]
torchdiffeq>=0.2.5
h5py
scipy>=1.10
huggingface_hub
matplotlib>=3.7

[paper:platform_machine != "aarch64"]
mdtraj>=1.9

[viz]
matplotlib>=3.7
plotly>=5.0
