IssamLaradji / sps
Official code for the Stochastic Polyak step-size optimizer
☆137Updated 4 months ago
Related projects ⓘ
Alternatives and complementary repositories for sps
- Implements stochastic line search☆117Updated last year
- 🧀 Pytorch code for the Fromage optimiser.☆122Updated 3 months ago
- ☆76Updated 4 years ago
- DeepOBS: A Deep Learning Optimizer Benchmark Suite☆103Updated 10 months ago
- 👩 Pytorch and Jax code for the Madam optimiser.☆50Updated 3 years ago
- This repository contains the results for the paper: "Descending through a Crowded Valley - Benchmarking Deep Learning Optimizers"☆180Updated 3 years ago
- Codebase for Learning Invariances in Neural Networks☆94Updated 2 years ago
- Code for: Implicit Competitive Regularization in GANs☆114Updated 2 years ago
- Bayesianize: A Bayesian neural network wrapper in pytorch☆86Updated 5 months ago
- Loss Patterns of Neural Networks☆82Updated 3 years ago
- Torchélie is a set of utility functions, layers, losses, models, trainers and other things for PyTorch.☆111Updated this week
- Prescribed Generative Adversarial Networks☆143Updated 4 years ago
- ☆97Updated 2 years ago
- Python implementation of GLN in different frameworks☆95Updated 4 years ago
- A collection of code snippets for my PyTorch Lightning projects☆107Updated 3 years ago
- A library for evaluating representations.☆76Updated 2 years ago
- Last-layer Laplace approximation code examples☆80Updated 3 years ago
- PyTorch-SSO: Scalable Second-Order methods in PyTorch☆141Updated last year
- Drop-in replacement for any ResNet with a significantly reduced memory footprint and better representation capabilities☆208Updated 6 months ago
- Quick modules to turn regular Neural Networks to Bayesian Neural Networks with Dropout.☆35Updated 3 years ago
- Pytorch implementation of Variational Dropout Sparsifies Deep Neural Networks☆83Updated 2 years ago
- PyTorch implementation of Sinusodial Representation networks (SIREN)☆263Updated last year
- ☆233Updated last year
- ☆156Updated 4 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆104Updated 2 years ago
- Code to accompany the paper Radial Bayesian Neural Networks: Beyond Discrete Support In Large-Scale Bayesian Deep Learning☆33Updated 4 years ago
- Explores the ideas presented in Deep Ensembles: A Loss Landscape Perspective (https://arxiv.org/abs/1912.02757) by Stanislav Fort, Huiyi …☆62Updated 4 years ago
- Implements sharpness-aware minimization (https://arxiv.org/abs/2010.01412) in TensorFlow 2.☆60Updated 2 years ago
- Cyclical Stochastic Gradient MCMC for Bayesian Deep Learning☆91Updated 4 years ago