tensordict
torch
torchaudio
torchvision