jxbz / fromage
🧀 Pytorch code for the Fromage optimiser.
☆123Updated 7 months ago
Alternatives and similar repositories for fromage:
Users that are interested in fromage are comparing it to the libraries listed below
- 👩 Pytorch and Jax code for the Madam optimiser.☆51Updated 4 years ago
- ☆98Updated 3 years ago
- Codebase for Learning Invariances in Neural Networks☆93Updated 2 years ago
- ☆153Updated 4 years ago
- A library for evaluating representations.☆74Updated 3 years ago
- Experiment code for "Randomized Automatic Differentiation"☆66Updated 4 years ago
- Code for: Implicit Competitive Regularization in GANs☆114Updated 3 years ago
- Code for the Thermodynamic Variational Objective☆26Updated 2 years ago
- ☆77Updated 4 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆106Updated 2 years ago
- Very deep VAEs in JAX/Flax☆46Updated 3 years ago
- Official code for the Stochastic Polyak step-size optimizer☆138Updated 7 months ago
- Pytorch implementation of the Power Spherical distribution☆74Updated 6 months ago
- Memory efficient MAML using gradient checkpointing☆83Updated 5 years ago
- Normalizing Flows in Jax☆107Updated 4 years ago
- Experiments for Meta-Learning Symmetries by Reparameterization☆56Updated 3 years ago
- This repository contains the results for the paper: "Descending through a Crowded Valley - Benchmarking Deep Learning Optimizers"☆180Updated 3 years ago
- ☆67Updated last year
- Padé Activation Units: End-to-end Learning of Activation Functions in Deep Neural Network☆63Updated 3 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 4 years ago
- DeepOBS: A Deep Learning Optimizer Benchmark Suite☆103Updated last year
- Mixture Density Networks (Bishop, 1994) tutorial in JAX☆58Updated 4 years ago
- TensorFlow implementation of "noisy K-FAC" and "noisy EK-FAC".☆60Updated 6 years ago
- Pytorch implementation of Variational Dropout Sparsifies Deep Neural Networks☆84Updated 3 years ago
- Drop-in replacement for any ResNet with a significantly reduced memory footprint and better representation capabilities☆209Updated 9 months ago
- Prescribed Generative Adversarial Networks☆143Updated 4 years ago
- TBA☆75Updated 5 years ago
- Easy-to-use AdaHessian optimizer (PyTorch)☆77Updated 4 years ago
- Loss Patterns of Neural Networks☆83Updated 3 years ago
- ☆47Updated 2 years ago