numpy
torch
torchvision
torcheval
tqdm
tensorboard
einops
axial_positional_embedding
