g-benton / loss-surface-simplexes
☆98Updated 3 years ago
Alternatives and similar repositories for loss-surface-simplexes:
Users that are interested in loss-surface-simplexes are comparing it to the libraries listed below
- Hessian spectral density estimation in TF and Jax☆121Updated 4 years ago
- 🧀 Pytorch code for the Fromage optimiser.☆123Updated 7 months ago
- A library for evaluating representations.☆74Updated 3 years ago
- Codebase for Learning Invariances in Neural Networks☆93Updated 2 years ago
- ☆29Updated 4 years ago
- Code for the Thermodynamic Variational Objective☆26Updated 2 years ago
- A Machine Learning workflow for Slurm.☆149Updated 4 years ago
- Drop-in replacement for any ResNet with a significantly reduced memory footprint and better representation capabilities☆209Updated 9 months ago
- Supporting code for the paper "Dangers of Bayesian Model Averaging under Covariate Shift"☆33Updated 2 years ago
- Code for the article "What if Neural Networks had SVDs?", to be presented as a spotlight paper at NeurIPS 2020.☆72Updated 6 months ago
- PyTorch-SSO: Scalable Second-Order methods in PyTorch☆145Updated last year
- ☆53Updated 6 months ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆106Updated 2 years ago
- Monotone operator equilibrium networks☆51Updated 4 years ago
- Code for the paper: "Tensor Programs II: Neural Tangent Kernel for Any Architecture"☆104Updated 4 years ago
- ☆156Updated 2 years ago
- ☆164Updated 6 months ago
- Bayesianize: A Bayesian neural network wrapper in pytorch☆88Updated 9 months ago
- Differentiable Sorting Networks☆110Updated last year
- Cyclical Stochastic Gradient MCMC for Bayesian Deep Learning☆93Updated 4 years ago
- Discrete Normalizing Flows implemented in PyTorch☆109Updated 3 years ago
- ☆34Updated 3 years ago
- ☆18Updated last year
- Normalizing Flows in Jax☆107Updated 4 years ago
- ☆153Updated 4 years ago
- ☆49Updated 4 years ago
- Riemannian Convex Potential Maps☆67Updated last year
- Experiment code for "Randomized Automatic Differentiation"☆66Updated 4 years ago
- Easy-to-use AdaHessian optimizer (PyTorch)☆77Updated 4 years ago
- This repository contains the results for the paper: "Descending through a Crowded Valley - Benchmarking Deep Learning Optimizers"☆180Updated 3 years ago