jsonargparse
matplotlib
mlflow
numpy<2
pandas
parse
pycocotools
pydot
pyyaml
scikit-learn
scipy
seaborn
tabulate
torch[opt-einsum]
torchmetrics
torchvision
tqdm
