torch
transformers
tiktoken
sentencepiece