mlflow==2.12.2
torch==2.2.1
torchvision==0.17.1