Mode Connectivity and Fast Geometric Ensembles in PyTorch
☆284Oct 24, 2022Updated 3 years ago
Alternatives and similar repositories for dnn-mode-connectivity
Users that are interested in dnn-mode-connectivity are comparing it to the libraries listed below
Sorting:
- PyTorch AutoNEB implementation to identify minimum energy paths, e.g. in neural network loss landscapes☆56Sep 14, 2022Updated 3 years ago
- Code Repo for "Subspace Inference for Bayesian Deep Learning"☆83Jun 17, 2024Updated last year
- Stochastic Weight Averaging in PyTorch☆977Aug 1, 2021Updated 4 years ago
- Code for visualizing the loss landscape of neural nets☆3,154Apr 5, 2022Updated 3 years ago
- Public Codebase for Rethinking Parameter Counting: Effective Dimensionality Revisited☆37Dec 27, 2022Updated 3 years ago
- ☆45Nov 4, 2020Updated 5 years ago
- Code repo for "A Simple Baseline for Bayesian Uncertainty in Deep Learning"☆478Jul 6, 2023Updated 2 years ago
- Codes for the paper "Optimizing Mode Connectivity via Neuron Alignment" from NeurIPS 2020.☆16Dec 10, 2020Updated 5 years ago
- ☆100Dec 8, 2021Updated 4 years ago
- PyHessian is a Pytorch library for second-order based analysis and training of Neural Networks☆778Jul 10, 2025Updated 7 months ago
- Efficient PyTorch Hessian eigendecomposition tools!☆386Feb 29, 2024Updated 2 years ago
- ☆23Nov 1, 2022Updated 3 years ago
- This repository provides code source used in the paper: A Mean Field Theory of Quantized Deep Networks: The Quantization-Depth Trade-Off☆13May 30, 2019Updated 6 years ago
- ☆252Dec 27, 2022Updated 3 years ago
- ☆133Mar 23, 2021Updated 4 years ago
- Code for ICML 2018 paper on "Fast and Scalable Bayesian Deep Learning by Weight-Perturbation in Adam" by Khan, Nielsen, Tangkaratt, Lin, …☆112Dec 19, 2018Updated 7 years ago
- Code for "EigenDamage: Structured Pruning in the Kronecker-Factored Eigenbasis" https://arxiv.org/abs/1905.05934☆113Mar 3, 2020Updated 6 years ago
- DiWA: Diverse Weight Averaging for Out-of-Distribution Generalization☆31Jan 31, 2023Updated 3 years ago
- Fast and Easy Infinite Neural Networks in Python☆2,377Mar 1, 2024Updated 2 years ago
- Improving Consistency-Based Semi-Supervised Learning with Weight Averaging☆189Feb 3, 2019Updated 7 years ago
- Code for Depth-wise Separable Convolutions: Performance Investigations☆19Jan 28, 2020Updated 6 years ago
- Code accompanying our paper "Finding trainable sparse networks through Neural Tangent Transfer" to be published at ICML-2020.☆13Jun 14, 2020Updated 5 years ago
- Implementation of the Deep Frank-Wolfe Algorithm -- Pytorch☆62Mar 6, 2021Updated 5 years ago
- ☆83Jan 15, 2020Updated 6 years ago
- ☆23Nov 24, 2018Updated 7 years ago
- Code to accompany paper 'Bayesian Deep Ensembles via the Neural Tangent Kernel'☆26Mar 12, 2021Updated 4 years ago
- This repository contains notebook implementations of the following Neural Process variants: Conditional Neural Processes (CNPs), Neural P…☆1,015Jan 19, 2021Updated 5 years ago
- Implementation of MAML in numpy, deriving gradients and implementing backprop manually☆14Nov 15, 2018Updated 7 years ago
- ☆15Apr 19, 2020Updated 5 years ago
- Implementation of Methods Proposed in Preventing Gradient Attenuation in Lipschitz Constrained Convolutional Networks (NeurIPS 2019)☆36Jun 25, 2020Updated 5 years ago
- BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.☆606Nov 28, 2025Updated 3 months ago
- This repo contains the code used for NeurIPS 2019 paper "Asymmetric Valleys: Beyond Sharp and Flat Local Minima".☆14Oct 25, 2019Updated 6 years ago
- Stochastic Gradient Langevin Dynamics for Bayesian learning☆36Nov 29, 2021Updated 4 years ago
- This is the implementation for CVPR 2022 Oral paper "Better Trigger Inversion Optimization in Backdoor Scanning."☆24Apr 5, 2022Updated 3 years ago
- Code for paper "SWALP: Stochastic Weight Averaging forLow-Precision Training".☆62May 11, 2019Updated 6 years ago
- Deep neural network kernel for Gaussian process☆211Sep 8, 2020Updated 5 years ago
- Corruption and Perturbation Robustness (ICLR 2019)☆1,138Aug 24, 2022Updated 3 years ago
- Implementation for the Lookahead Optimizer.☆244Apr 29, 2022Updated 3 years ago
- Bayesian Deep Learning Benchmarks☆672Mar 24, 2023Updated 2 years ago