torch
accelerate
torchvision
bitsandbytes
transformers