tfjgeorge / nngeometry-examplesLinks
Example code for the NNGeometry PyTorch library
☆10Updated 3 months ago
Alternatives and similar repositories for nngeometry-examples
Users that are interested in nngeometry-examples are comparing it to the libraries listed below
Sorting:
- PyTorch implementation of FIM and empirical FIM☆60Updated 7 years ago
- Deep Learning & Information Bottleneck☆62Updated 2 years ago
- Code for Knowledge-Adaptation Priors based on the NeurIPS 2021 paper by Khan and Swaroop.☆16Updated 3 years ago
- Federated posterior averaging implemented in JAX☆52Updated 2 years ago
- Code to implement the AND-mask and geometric mean to do gradient based optimization, from the paper "Learning explanations that are hard …☆41Updated 5 years ago
- PyTorch implementation of the paper "Continuous Wasserstein-2 Barycenter Estimation without Minimax Optimization" (ICLR 2021)☆25Updated 3 years ago
- Tilted Empirical Risk Minimization (ICLR '21)☆60Updated 2 years ago
- Distilling Model Failures as Directions in Latent Space☆47Updated 2 years ago
- ModelDiff: A Framework for Comparing Learning Algorithms☆58Updated 2 years ago
- Implementation of the models and datasets used in "An Information-theoretic Approach to Distribution Shifts"☆25Updated 4 years ago
- PyTorch Implementation of Reptile☆21Updated 4 years ago
- ☆38Updated 4 years ago
- Implementations of orthogonal and semi-orthogonal convolutions in the Fourier domain with applications to adversarial robustness☆48Updated 4 years ago
- A modern look at the relationship between sharpness and generalization [ICML 2023]☆43Updated 2 years ago
- Computationally friendly hyper-parameter search with DP-SGD☆25Updated 11 months ago
- Post-processing for fair classification☆16Updated 5 months ago
- ☆18Updated 3 years ago
- A simple Jax implementation of influence functions.☆18Updated last year
- The Full Spectrum of Deepnet Hessians at Scale: Dynamics with SGD Training and Sample Size☆17Updated 6 years ago
- Diagnosing Vulnerability of Variational Auto-Encoders to Adversarial Attacks☆13Updated 3 years ago
- ☆28Updated 2 years ago
- Towards Understanding Sharpness-Aware Minimization [ICML 2022]☆36Updated 3 years ago
- Pytorch Implementation of the Nonlinear Information Bottleneck☆41Updated last year
- Distributional and Outlier Robust Optimization (ICML 2021)☆28Updated 4 years ago
- Energy Based Models are a quite novel technique for density estimation. In this university project I explore this new research topic and …☆16Updated 4 years ago
- ☆10Updated 3 years ago
- Code for the ICLR 2021 Paper "In-N-Out: Pre-Training and Self-Training using Auxiliary Information for Out-of-Distribution Robustness"☆13Updated 4 years ago
- Code for Environment Inference for Invariant Learning (ICML 2021 Paper)☆51Updated 4 years ago
- Dynamic causal Bayesian optimisation☆40Updated 2 years ago
- ☆14Updated 5 years ago