tfjgeorge / nngeometry-examplesLinks
Example code for the NNGeometry PyTorch library
☆10Updated 5 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:
- ModelDiff: A Framework for Comparing Learning Algorithms☆58Updated 2 years ago
- Deep Learning & Information Bottleneck☆63Updated 2 years ago
- ☆28Updated 2 years ago
- Code for Knowledge-Adaptation Priors based on the NeurIPS 2021 paper by Khan and Swaroop.☆16Updated 4 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 FIM and empirical FIM☆60Updated 7 years ago
- Towards Understanding Sharpness-Aware Minimization [ICML 2022]☆38Updated 3 years ago
- The Full Spectrum of Deepnet Hessians at Scale: Dynamics with SGD Training and Sample Size☆19Updated 6 years ago
- ☆38Updated 2 years ago
- A modern look at the relationship between sharpness and generalization [ICML 2023]☆43Updated 2 years ago
- ☆10Updated 3 years ago
- PyTorch implementation of the paper "Continuous Wasserstein-2 Barycenter Estimation without Minimax Optimization" (ICLR 2021)☆33Updated 3 years ago
- Diagnosing Vulnerability of Variational Auto-Encoders to Adversarial Attacks☆13Updated 3 years ago
- ☆38Updated 4 years ago
- Implementation of the models and datasets used in "An Information-theoretic Approach to Distribution Shifts"☆25Updated 4 years ago
- A simple Jax implementation of influence functions.☆20Updated last year
- Code for the paper: "Tensor Programs II: Neural Tangent Kernel for Any Architecture"☆109Updated 5 years ago
- Computationally friendly hyper-parameter search with DP-SGD☆25Updated last year
- Source code of "What can linearized neural networks actually say about generalization?☆20Updated 4 years ago
- Code for Accelerated Linearized Laplace Approximation for Bayesian Deep Learning (ELLA, NeurIPS 22')☆17Updated 3 years ago
- ☆18Updated 3 years ago
- Official code for "In Search of Robust Measures of Generalization" (NeurIPS 2020)☆28Updated 5 years ago
- [NeurIPS 2021] A Geometric Analysis of Neural Collapse with Unconstrained Features☆60Updated 3 years ago
- Federated posterior averaging implemented in JAX☆53Updated 2 years ago
- Distributional and Outlier Robust Optimization (ICML 2021)☆28Updated 4 years ago
- DiWA: Diverse Weight Averaging for Out-of-Distribution Generalization☆31Updated 3 years ago
- Explores the ideas presented in Deep Ensembles: A Loss Landscape Perspective (https://arxiv.org/abs/1912.02757) by Stanislav Fort, Huiyi …☆66Updated 5 years ago
- Tilted Empirical Risk Minimization (ICLR '21)☆60Updated 2 years ago
- (ICML 2023) Feature learning in deep classifiers through Intermediate Neural Collapse: Accompanying code☆16Updated 2 years ago
- Implementations of orthogonal and semi-orthogonal convolutions in the Fourier domain with applications to adversarial robustness☆48Updated 4 years ago