matplotlib
pyarrow
torch
torchaudio
torchvision
wandb

# Optional dependencies
# jupyter
# pandas