torch
torchvision
numpy
einops
performer-pytorch
linformer
