Official code: Learning Expressive Priors for Generalization and Uncertainty Estimation in Neural Networks - View it on GitHub