google-deepmind / conformal_training
This repository contains a Jax implementation of conformal training corresponding to the ICLR'22 paper "learning optimal conformal classifiers".
☆129Updated 2 years ago
Alternatives and similar repositories for conformal_training:
Users that are interested in conformal_training are comparing it to the libraries listed below
- Conformal prediction for controlling monotonic risk functions. Simple accompanying PyTorch code for conformal risk control in computer vi…☆62Updated 2 years ago
- A library for uncertainty quantification based on PyTorch☆121Updated 3 years ago
- Learn then Test: Calibrating Predictive Algorithms to Achieve Risk Control☆66Updated 3 months ago
- Materials of the Nordic Probabilistic AI School 2023.☆89Updated last year
- ☆50Updated 3 weeks ago
- Bayesianize: A Bayesian neural network wrapper in pytorch☆88Updated 9 months ago
- A package for conformal prediction with conditional guarantees.☆50Updated 3 weeks ago
- relplot: Utilities for measuring calibration and plotting reliability diagrams☆138Updated 8 months ago
- Official codebase for "Distribution-Free, Risk-Controlling Prediction Sets"☆87Updated last year
- A primer on Bayesian Neural Networks. The aim of this reading list is to facilitate the entry of new researchers into the field of Bayesi…☆52Updated last year
- A statistical toolkit for scientific discovery using machine learning☆73Updated 7 months ago
- Training and evaluating NBM and SPAM for interpretable machine learning.☆77Updated last year
- Laplace Redux -- Effortless Bayesian Deep Learning☆41Updated last year
- Simple (and cheap!) neural network uncertainty estimation☆61Updated last week
- Framework code with wandb, checkpointing, logging, configs, experimental protocols. Useful for fine-tuning models or training from scratc…☆148Updated 2 years ago
- Wrapper for a PyTorch classifier which allows it to output prediction sets. The sets are theoretically guaranteed to contain the true cla…☆235Updated 2 years ago
- Parameter-Free Optimizers for Pytorch☆113Updated 9 months ago
- Code for the paper "Bayesian Neural Network Priors Revisited"☆56Updated 3 years ago
- ☆146Updated last year
- Agustinus' very opiniated publication-ready plotting library☆61Updated 2 weeks ago
- Materials of the Nordic Probabilistic AI School 2022.☆174Updated 2 years ago
- Euclidean Wasserstein-2 optimal transportation☆44Updated last year
- ☆18Updated last year
- Supporing code for the paper "Bayesian Model Selection, the Marginal Likelihood, and Generalization".☆35Updated 2 years ago
- Official repository for CMU Machine Learning Department's 10732: Robustness and Adaptivity in Shifting Environments☆73Updated 2 years ago
- ☆108Updated 2 years ago
- Materials for STAT 991: Topics In Modern Statistical Learning (UPenn, 2022 Spring) - uncertainty quantification, conformal prediction, ca…☆166Updated 9 months ago
- PyTorch linear operators for curvature matrices (Hessian, Fisher/GGN, KFAC, ...)☆31Updated this week
- A package for statistically rigorous scientific discovery using machine learning. Implements prediction-powered inference.☆222Updated last month
- Official Implementation of "Transformers Can Do Bayesian Inference", the PFN paper☆201Updated 3 months ago