krishnap25 / geom_median
Fast and differentiable geometric median, a multivariate median analogue. Install with `pip install geom-median`
☆49Updated 3 years ago
Alternatives and similar repositories for geom_median:
Users that are interested in geom_median are comparing it to the libraries listed below
- Source code of "What can linearized neural networks actually say about generalization?☆20Updated 3 years ago
- Deep Networks Grok All the Time and Here is Why☆34Updated 11 months ago
- PyTorch implementation of the paper "Continuous Wasserstein-2 Barycenter Estimation without Minimax Optimization" (ICLR 2021)☆23Updated 2 years ago
- PyTorch linear operators for curvature matrices (Hessian, Fisher/GGN, KFAC, ...)☆36Updated this week
- Learn then Test: Calibrating Predictive Algorithms to Achieve Risk Control☆66Updated 5 months ago
- SGD with large step sizes learns sparse features [ICML 2023]☆32Updated last year
- Official Implementation of the paper: "A Rate-Distorion View of Uncertainty Quantification", ICML 2024☆28Updated 7 months ago
- Large-scale uncertainty benchmark in deep learning.☆56Updated 3 months ago
- For calculating Shapley values via linear regression.☆67Updated 3 years ago
- Conformal prediction for controlling monotonic risk functions. Simple accompanying PyTorch code for conformal risk control in computer vi…☆66Updated 2 years ago
- "How to Trust Your Diffusion Models: A Convex Optimization Approach to Conformal Risk Control"☆17Updated last month
- Code for the paper "Getting a CLUE: A Method for Explaining Uncertainty Estimates"☆35Updated 11 months ago
- Fast hyperparameter settings for non-smooth estimators:☆40Updated last year
- ☆26Updated 3 months ago
- Influence Estimation for Gradient-Boosted Decision Trees☆27Updated 10 months ago
- Code for 'Periodic Activation Functions Induce Stationarity' (NeurIPS 2021)☆18Updated 3 years ago
- A Python package for intrinsic dimension estimation☆86Updated 2 months ago
- TorchDR - PyTorch Dimensionality Reduction☆105Updated 2 months ago
- Pytorch code for "Improving Self-Supervised Learning by Characterizing Idealized Representations"☆40Updated 2 years ago
- Code for PAC-Bayes Compression Bounds So Tight That They Can Explain Generalization, NeurIPS 2022☆15Updated 2 years ago
- An amortized approach for calculating local Shapley value explanations☆97Updated last year
- ☆10Updated 3 years ago
- Deep Learning & Information Bottleneck☆59Updated last year
- Welcome to Uncertainty Metrics! The goal of this library is to provide an easy-to-use interface for both measuring uncertainty across Goo…☆22Updated 4 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
- AutoML Two-Sample Test☆19Updated 2 years ago
- Wrap around any model to output differentially private prediction sets with finite sample validity on any dataset.☆17Updated last year
- A package for conformal prediction with conditional guarantees.☆53Updated 2 months ago
- The Linear Optimal Transport Framework☆14Updated 4 years ago
- Agustinus' very opiniated publication-ready plotting library☆63Updated 2 months ago