torch
transformers
lightning
lightning-bolts
numpy
pytorch-ignite
pytorch-lightning
pytorch-lightning-bolts
pytorch-warmup
scikit-learn
scipy
seaborn
gitpython