gitpython
numpy
onnx
tensordict
torch
torchvision