setuptools
build
numpy
scipy
torch
pytest
tqdm
pyyaml
colossus
optuna
