krishnap25 / geom_median
Fast and differentiable geometric median, a multivariate median analogue. Install with `pip install geom-median`
☆42Updated 2 years ago
Related projects ⓘ
Alternatives and complementary repositories for geom_median
- Code for the paper "Getting a CLUE: A Method for Explaining Uncertainty Estimates"☆36Updated 6 months ago
- PyTorch implementation of the paper "Continuous Wasserstein-2 Barycenter Estimation without Minimax Optimization" (ICLR 2021)☆23Updated 2 years ago
- Adaptive, interpretable wavelets across domains (NeurIPS 2021)☆71Updated 2 years ago
- ☆141Updated last year
- Pytorch code for "Improving Self-Supervised Learning by Characterizing Idealized Representations"☆40Updated last year
- Transformers with doubly stochastic attention☆40Updated 2 years ago
- A Python package for intrinsic dimension estimation☆81Updated 3 months ago
- Conformal prediction for controlling monotonic risk functions. Simple accompanying PyTorch code for conformal risk control in computer vi…☆59Updated last year
- Instructions and examples to deploy some PyTorch code on slurm using a Singularity Container☆32Updated last year
- SGD with large step sizes learns sparse features [ICML 2023]☆32Updated 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…☆49Updated last year
- Supporing code for the paper "Bayesian Model Selection, the Marginal Likelihood, and Generalization".☆35Updated 2 years ago
- A fast hierarchical dimensionality reduction algorithm.☆62Updated 3 weeks ago
- A library for uncertainty quantification based on PyTorch☆119Updated 2 years ago
- A package for conformal prediction with conditional guarantees.☆47Updated 4 months ago
- ☆109Updated 2 years ago
- Implementations of orthogonal and semi-orthogonal convolutions in the Fourier domain with applications to adversarial robustness☆42Updated 3 years ago
- Official Implementation of the paper: "A Rate-Distorion View of Uncertainty Quantification", ICML 2024☆28Updated 2 months ago
- This repository contains a Jax implementation of conformal training corresponding to the ICLR'22 paper "learning optimal conformal classi…☆121Updated 2 years ago
- Code of "Deep invariant networks with differentiable augmentation layers"☆18Updated 2 years ago
- Uncertainty-aware representation learning (URL) benchmark☆98Updated 8 months ago
- AutoML Two-Sample Test☆19Updated 2 years ago
- ☆97Updated 6 months ago
- This repository includes code to reproduce the tables in "Loss Landscapes are All You Need: Neural Network Generalization Can Be Explaine…☆34Updated last year
- scipy linear operators for the Hessian, Fisher/GGN, and more in PyTorch☆18Updated 2 weeks ago
- Model Zoos published at the NeurIPS 2022 Dataset & Benchmark track: "Model Zoos: A Dataset of Diverse Populations of Neural Network Model…☆54Updated last year
- Agustinus' very opiniated publication-ready plotting library☆58Updated 2 months ago
- Model-agnostic posthoc calibration without distributional assumptions☆42Updated last year
- ☆36Updated 2 years ago
- ☆23Updated 2 years ago