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:
- Code for Knowledge-Adaptation Priors based on the NeurIPS 2021 paper by Khan and Swaroop.☆16Updated 4 years ago
- PyTorch implementation of FIM and empirical FIM☆60Updated 7 years ago
- ModelDiff: A Framework for Comparing Learning Algorithms☆58Updated 2 years ago
- ☆10Updated 3 years ago
- Federated posterior averaging implemented in JAX☆53Updated 2 years ago
- ☆28Updated 2 years ago
- Deep Learning & Information Bottleneck☆63Updated 2 years ago
- ☆38Updated 4 years ago
- A modern look at the relationship between sharpness and generalization [ICML 2023]☆43Updated 2 years ago
- The Full Spectrum of Deepnet Hessians at Scale: Dynamics with SGD Training and Sample Size☆19Updated 6 years ago
- Post-processing for fair classification☆16Updated 7 months ago
- Code for Accelerated Linearized Laplace Approximation for Bayesian Deep Learning (ELLA, NeurIPS 22')☆17Updated 3 years ago
- Computationally friendly hyper-parameter search with DP-SGD☆25Updated last year
- Towards Understanding Sharpness-Aware Minimization [ICML 2022]☆38Updated 3 years ago
- MDL Complexity computations and experiments from the paper "Revisiting complexity and the bias-variance tradeoff".☆18Updated 2 years ago
- Dynamic causal Bayesian optimisation☆40Updated 2 years ago
- Official implementation for Training Certifiably Robust Neural Networks with Efficient Local Lipschitz Bounds (NeurIPS, 2021).☆25Updated 3 years ago
- Code for the paper: "Tensor Programs II: Neural Tangent Kernel for Any Architecture"☆109Updated 5 years ago
- ☆14Updated 5 years ago
- Distributional and Outlier Robust Optimization (ICML 2021)☆28Updated 4 years ago
- Official code for "In Search of Robust Measures of Generalization" (NeurIPS 2020)☆28Updated 5 years ago
- Diagnosing Vulnerability of Variational Auto-Encoders to Adversarial Attacks☆13Updated 3 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
- 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
- Source code of "What can linearized neural networks actually say about generalization?☆20Updated 4 years ago
- Distilling Model Failures as Directions in Latent Space☆47Updated 3 years ago
- Code for the paper "Getting a CLUE: A Method for Explaining Uncertainty Estimates"☆35Updated last year
- (ICML 2023) Feature learning in deep classifiers through Intermediate Neural Collapse: Accompanying code☆16Updated 2 years ago
- ☆34Updated 2 years ago
- ☆18Updated 3 years ago