torch
accelerate
transformers