krishnap25 / geom_median
Fast and differentiable geometric median, a multivariate median analogue. Install with `pip install geom-median`
☆45Updated 2 years ago
Alternatives and similar repositories for geom_median:
Users that are interested in geom_median 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
- Replicating and dissecting the git-re-basin project in one-click-replication Colabs☆36Updated 2 years ago
- PyTorch implementation of the paper "Continuous Wasserstein-2 Barycenter Estimation without Minimax Optimization" (ICLR 2021)☆23Updated 2 years ago
- Code for PAC-Bayes Compression Bounds So Tight That They Can Explain Generalization, NeurIPS 2022☆15Updated 2 years ago
- Parameter-Free Optimizers for Pytorch☆109Updated 9 months ago
- (ICML 2023) Feature learning in deep classifiers through Intermediate Neural Collapse: Accompanying code☆13Updated last year
- ☆46Updated last week
- Implementations of orthogonal and semi-orthogonal convolutions in the Fourier domain with applications to adversarial robustness☆43Updated 3 years ago
- Python package to fetch data from the LIBSVM website.☆19Updated last year
- Code for the paper "Getting a CLUE: A Method for Explaining Uncertainty Estimates"☆35Updated 9 months ago
- Wrap around any model to output differentially private prediction sets with finite sample validity on any dataset.☆17Updated 10 months ago
- SGD with large step sizes learns sparse features [ICML 2023]☆32Updated last year
- For calculating Shapley values via linear regression.☆66Updated 3 years ago
- Official Implementation of the paper: "A Rate-Distorion View of Uncertainty Quantification", ICML 2024☆28Updated 4 months ago
- Code of the paper "Beyond calibration: estimating the grouping loss of modern neural networks" published in ICLR 2023.☆11Updated last year
- ☆143Updated last year
- ☆25Updated last week
- CHOP: An optimization library based on PyTorch, with applications to adversarial examples and structured neural network training.☆77Updated 10 months ago
- A package for conformal prediction with conditional guarantees.☆49Updated 6 months ago
- Code for 'Periodic Activation Functions Induce Stationarity' (NeurIPS 2021)☆18Updated 3 years ago
- Training and evaluating NBM and SPAM for interpretable machine learning.☆77Updated last year
- This repository contains a Jax implementation of conformal training corresponding to the ICLR'22 paper "learning optimal conformal classi…☆128Updated 2 years ago
- 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
- Transformers with doubly stochastic attention☆44Updated 2 years ago
- A project to improve out-of-distribution detection (open set recognition) and uncertainty estimation by changing a few lines of code in y…☆45Updated 2 years ago
- Code for "The Intrinsic Dimension of Images and Its Impact on Learning" - ICLR 2021 Spotlight https://openreview.net/forum?id=XJk19XzGq2J☆65Updated 9 months ago
- Learn then Test: Calibrating Predictive Algorithms to Achieve Risk Control☆65Updated 2 months ago
- This code is no longer maintained. The codebase has been moved to https://github.com/scikit-learn-contrib/skglm. This repository only ser…☆17Updated 2 years ago
- ModelDiff: A Framework for Comparing Learning Algorithms☆54Updated last year
- Conformal Prediction + Federated Learning☆10Updated 10 months ago