torch>=1.1.0
torchvision
scipy
numpy
pytest
editdistance
dataclasses
rich
scikit-learn
wandb
submitit
