alembic==1.4.1
mlflow==1.23.1
numpy==1.21
torch==1.9.0
pytorch-lightning==1.6.0
wandb==0.12.7