matplotlib 
pyarrow 
torch 
torchaudio 
torchvision 
wandb

# Optional dependencies
# jupyter
# matplotlib