jsonargparse
matplotlib
mlflow
numpy
nvidia-dali-cuda130
pandas
parse
pycocotools
pydot
pyyaml
scikit-learn
scipy
seaborn
tabulate
torch>=2
torchmetrics
torchvision
tqdm
