numpy>=1.17.2
pandas>=1.1.5
scikit-learn
torch>=1.7.0
wandb
einops
sphinx_rtd_theme
recommonmark