DanReia

import torch
from torch.distributions import constraints

import pyro
import pyro.distributions as dist
from pyro.infer import SVI, TraceEnum_ELBO, config_enumerate

import matplotlib.pyplot as plt
pyro.enable_validation(True)