accelerate
transformers
torch
numpy