packaging
ninja
einops>=0.5
matplotlib
pandas
torch>=2.1.0
wandb
transformers>=4.34.1
numpy>=1.14.2