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☆122Updated 4 years ago
- Codebase for Learning Invariances in Neural Networks☆94Updated 2 years ago
- Experiments for Meta-Learning Symmetries by Reparameterization☆56Updated 3 years ago
- 🧀 Pytorch code for the Fromage optimiser.☆123Updated 8 months ago
- ☆53Updated 7 months ago
- Discrete Normalizing Flows implemented in PyTorch☆109Updated 3 years ago
- Code for the Thermodynamic Variational Objective☆26Updated 2 years ago
- PyTorch-SSO: Scalable Second-Order methods in PyTorch☆145Updated last year
- ☆135Updated 3 years ago
- Code for the article "What if Neural Networks had SVDs?", to be presented as a spotlight paper at NeurIPS 2020.☆74Updated 7 months ago
- Autoregressive Energy Machines☆77Updated 2 years ago
- Monotone operator equilibrium networks☆51Updated 4 years ago
- A library for evaluating representations.☆76Updated 3 years ago
- Supporting code for the paper "Dangers of Bayesian Model Averaging under Covariate Shift"☆33Updated 2 years ago
- ☆34Updated 3 years ago
- Contains code for the NeurIPS 2020 paper by Pan et al., "Continual Deep Learning by FunctionalRegularisation of Memorable Past"☆44Updated 4 years ago
- ☆166Updated 7 months ago
- Hypergradient descent☆145Updated 9 months ago
- Riemannian Convex Potential Maps☆67Updated last year
- Bayesianize: A Bayesian neural network wrapper in pytorch☆88Updated 10 months ago
- Convolutional Neural Tangent Kernel☆109Updated 5 years ago
- A Machine Learning workflow for Slurm.☆149Updated 4 years ago
- Public Codebase for Rethinking Parameter Counting: Effective Dimensionality Revisited☆37Updated 2 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆108Updated 2 years ago
- Computing the eigenvalues of Neural Tangent Kernel and Conjugate Kernel (aka NNGP kernel) over the boolean cube☆48Updated 5 years ago
- Reparameterize your PyTorch modules☆70Updated 4 years ago
- Differentiable Sorting Networks☆112Updated last year
- Normalizing Flows in Jax☆107Updated 4 years ago
- Normalizing Flows using JAX☆82Updated last year
- CHOP: An optimization library based on PyTorch, with applications to adversarial examples and structured neural network training.☆77Updated last year