drake
hydra-core
torch
pytest
matplotlib
numpy
scipy
tqdm
wandb
scikit-learn
omegaconf
