torch
tokenizers
transformers
accelerate>=1.1.0