torch
fastcore
numpy
matplotlib
torcheval
torchvision
fastprogress
einops
accelerate
