torch
numpy
scipy
matplotlib
wandb
pyyaml
arithmetic_compressor
pycallgraph2
setuptools
wheel