TorchUncertainty

This is the official Hugging Face account for the TorchUncertainty library. You will find its pre-trained weights here and several other models and datasets.

More specifically, the datasets contain the models trained for A Symmetry-Aware Exploration of Bayesian Neural Network Posteriors.