tfjgeorge / nngeometry-examples
Example code for the NNGeometry PyTorch library
☆10Updated 2 years ago
Related projects ⓘ
Alternatives and complementary repositories for nngeometry-examples
- Deep Learning & Information Bottleneck☆50Updated last year
- Code for Knowledge-Adaptation Priors based on the NeurIPS 2021 paper by Khan and Swaroop.☆16Updated 2 years ago
- Distributional and Outlier Robust Optimization (ICML 2021)☆27Updated 3 years ago
- ☆23Updated last year
- ☆17Updated 2 years ago
- (ICML 2023) Feature learning in deep classifiers through Intermediate Neural Collapse: Accompanying code☆13Updated last year
- ☆25Updated last year
- Source code of "What can linearized neural networks actually say about generalization?☆18Updated 3 years ago
- [NeurIPS 2021] A Geometric Analysis of Neural Collapse with Unconstrained Features☆53Updated 2 years ago
- Energy-Based Models for Continual Learning Official Repository (PyTorch)☆40Updated 2 years ago
- A simple Jax implementation of influence functions.☆16Updated 7 months ago
- Continual learning of task-specific approximations of the parameter posterior distribution via a shared hypernetwork.☆16Updated 3 weeks ago
- Code for NeurIPS'23 paper "A Bayesian Approach To Analysing Training Data Attribution In Deep Learning"☆14Updated 10 months ago
- Federated posterior averaging implemented in JAX☆49Updated last year
- ☆10Updated 2 years ago
- Official implementation of the paper "Interventions, Where and How? Experimental Design for Causal Models at Scale", NeurIPS 2022.☆19Updated last year
- A modern look at the relationship between sharpness and generalization [ICML 2023]☆43Updated last year
- Stochastic Gradient Langevin Dynamics for Bayesian learning☆30Updated 2 years ago
- [ICML2022] Training Your Sparse Neural Network Better with Any Mask. Ajay Jaiswal, Haoyu Ma, Tianlong Chen, ying Ding, and Zhangyang Wang☆26Updated 2 years ago
- PyTorch implementation of FIM and empirical FIM☆58Updated 6 years ago
- ☆26Updated last year
- ☆41Updated last year
- Bayesian Deep Learning Library☆11Updated last year
- ☆16Updated last year
- Code for testing DCT plus Sparse (DCTpS) networks☆14Updated 3 years ago
- Code and results accompanying our paper titled RLSbench: Domain Adaptation under Relaxed Label Shift☆35Updated last year
- ModelDiff: A Framework for Comparing Learning Algorithms☆53Updated last year
- Code for Accelerated Linearized Laplace Approximation for Bayesian Deep Learning (ELLA, NeurIPS 22')☆16Updated 2 years ago
- ☆23Updated 2 years ago
- ☆21Updated last year