torch
accelerate
transformers
numpy