numpy
pandas
torch
scikit-learn
pyyaml

[dev]
pytest
build

[experiments]
matplotlib
scipy
statsmodels
