torch

transformers

tiktoken