torch
torchvision
torchmetrics
numpy
timm
einops
opencv-python
dill