torch
torchvision
transformers
datasets==3.6.0
numpy
pandas
einops
opt_einsum
rotary_embedding_torch
wandb
pyyaml
tqdm
zstandard
