torch
transformers
einops
beartype
zetascale