torch==2.1.0
torchvision==0.16.0
numpy==1.26.1
pyyaml==6.0.1
pysdd==0.2.11
wandb==0.16.0
tqdm==4.66.1
scikit-learn==1.3.2