pytorch-ignite
pyyaml
mlflow
