torch
transformers
einops
opt_einsum
rotary_embedding_torch
wandb
pyyaml
