g-benton / loss-surface-simplexesLinks
☆99Updated 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
Sorting:
- 🧀 Pytorch code for the Fromage optimiser.☆124Updated 10 months ago
- Codebase for Learning Invariances in Neural Networks☆95Updated 2 years ago
- PyTorch-SSO: Scalable Second-Order methods in PyTorch☆146Updated last year
- Hessian spectral density estimation in TF and Jax☆123Updated 4 years ago
- ☆133Updated 4 years ago
- A library for evaluating representations.☆76Updated 3 years ago
- Last-layer Laplace approximation code examples☆83Updated 3 years ago
- Monotone operator equilibrium networks☆52Updated 4 years ago
- A Machine Learning workflow for Slurm.☆149Updated 4 years ago
- Easy-to-use AdaHessian optimizer (PyTorch)☆78Updated 4 years ago
- Hypergradient descent☆148Updated last year
- Experiments for Meta-Learning Symmetries by Reparameterization☆56Updated 4 years ago
- ☆53Updated 10 months ago
- Bayesianize: A Bayesian neural network wrapper in pytorch☆88Updated last year
- ☆34Updated 3 years ago
- Supporting code for the paper "Dangers of Bayesian Model Averaging under Covariate Shift"☆33Updated 2 years ago
- Differentiable Sorting Networks☆115Updated last year
- Code for the paper: "Implicit Bias of Gradient Descent for Wide Two-layer Neural Networks Trained with the Logistic Loss" (Chizat and Bac…☆7Updated 5 years ago
- Code for the paper: "Tensor Programs II: Neural Tangent Kernel for Any Architecture"☆105Updated 4 years ago
- ☆64Updated last year
- Riemannian Convex Potential Maps☆67Updated 2 years ago
- ☆153Updated 5 years ago
- Code to implement the AND-mask and geometric mean to do gradient based optimization, from the paper "Learning explanations that are hard …☆39Updated 4 years ago
- 👩 Pytorch and Jax code for the Madam optimiser.☆51Updated 4 years ago
- {KFAC,EKFAC,Diagonal,Implicit} Fisher Matrices and finite width NTKs in PyTorch☆214Updated last week
- ☆157Updated 2 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆110Updated 2 years ago
- Code for the Thermodynamic Variational Objective☆26Updated 2 years ago
- Convolutional Neural Tangent Kernel☆111Updated 5 years ago
- Stochastic Automatic Differentiation library for PyTorch.☆204Updated 9 months ago