torch
torchvision
perforatedai
wandb