torch
tqdm
onnx
onnxscript
torchinfo
datasets
