torch>=1.6.0
python-Levenshtein
numpy
pandas
astropy
sentencepiece
pytorch-lightning
hydra-core
wget
wandb
