torch
torchvision
torchmetrics
numpy
lion_pytorch
