torch
torchvision
transformers
numpy
einops
opt_einsum
rotary_embedding_torch
wandb
pyyaml
tqdm
