The module pyro.nn provides implementations of neural network modules that are useful in the context of deep probabilistic programming.
pyro.nn.module
pyro.nn.auto_reg_nn.AutoRegressiveNN
pyro.nn.dense_nn.DenseNN
pyro.nn.auto_reg_nn.ConditionalAutoRegressiveNN
pyro.nn.dense_nn.ConditionalDenseNN