torch>=2.0
numpy

[experiments]
pandas
tqdm
pyyaml

[test]
pytest
