torch
torchvision
einops
numpy
ftfy
regex
axial-positional-embedding
youtokentome
tokenizers