torch>2
requests
einops
rotary-embedding-torch
ml_collections
numpy
dm-tree
pyyaml
scipy
