Bayesian NN in torch

Is there a torch implementation of distribution layers, so it is possible to predict uncertainty in Neural Networks? Something similar to Tensorflow Probability or Yarin Gal's concrete dropout implementation.

Hi @fg.cordoni ,

We have support for a few distributions, and they can be use within nn_modules, here's an example: