einops>=0.4.1
torch>=1.10.0
rotary-embedding-torch>=0.1.0
torchvision>=0.11.1
colorama>=0.4.5
tqdm>=4.64.0
torchmetrics>=0.9.3
numpy>=1.22.0
matplotlib>=3.5.2
