torch
numpy
dataclasses
tiktoken