numpy==1.26.1
torch==2.1.0
pyyaml==6.0.1
# wandb==0.17.1
# pulp==2.8.0