torch>=0.4.1
torchvision>=0.2.1
argparse~=1.4.0
quadprog


numpy~=1.20.2
wandb~=0.11.2