torch
transformers
accelerate
tensorflow_probability
sentencepiece