numpy
torch
pyro-ppl<1.6
