[nabusman]

Or Pyro https://github.com/uber/pyro

[eli_gottlieb]

Or ProbTorch https://github.com/probtorch/probtorch