torch
torchvision
numpy
xformers
torchmetrics
lightning
pandas
wandb
einops
medmnist