sympy==1.12.1
torch
pytorch-lightning==1.9.5
deepspeed
hydra-core
datasets
miditok
wandb
tensorboard