jxbz / madam
👩 Pytorch and Jax code for the Madam optimiser.
☆50Updated 3 years ago
Related projects ⓘ
Alternatives and complementary repositories for madam
- 🧀 Pytorch code for the Fromage optimiser.☆122Updated 3 months ago
- Very deep VAEs in JAX/Flax☆45Updated 3 years ago
- ☆11Updated 4 years ago
- Pytorch implementation of Variational Dropout Sparsifies Deep Neural Networks☆83Updated 2 years ago
- A discrete sequential VAE☆38Updated 4 years ago
- ☆66Updated last year
- Code to accompany the paper Radial Bayesian Neural Networks: Beyond Discrete Support In Large-Scale Bayesian Deep Learning☆33Updated 4 years ago
- Pretrained TorchVision models on CIFAR10 dataset (with weights)☆24Updated 4 years ago
- Code for the Thermodynamic Variational Objective☆26Updated 2 years ago
- Implements stochastic line search☆117Updated last year
- Estimating Gradients for Discrete Random Variables by Sampling without Replacement☆39Updated 4 years ago
- ☆156Updated 4 years ago
- [AAAI 2020 Oral] Low-variance Black-box Gradient Estimates for the Plackett-Luce Distribution☆36Updated 3 years ago
- ☆45Updated 5 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 3 years ago
- ☆26Updated 5 years ago
- Mixture Density Networks (Bishop, 1994) tutorial in JAX☆58Updated 4 years ago
- ☆49Updated 4 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆104Updated 2 years ago
- Experiment code for "Randomized Automatic Differentiation"☆67Updated 4 years ago
- Autoregressive Energy Machines☆77Updated last year
- Pytorch implementation of preconditioned stochastic gradient descent (affine group preconditioner, low-rank approximation preconditioner …☆110Updated 3 weeks ago
- Structured matrices for compressing neural networks☆67Updated last year
- TensorFlow implementation of "noisy K-FAC" and "noisy EK-FAC".☆60Updated 5 years ago
- Normalizing Flows in Jax☆105Updated 4 years ago
- Monotone operator equilibrium networks☆51Updated 4 years ago
- Loss Patterns of Neural Networks☆82Updated 3 years ago
- Codebase for Learning Invariances in Neural Networks☆94Updated 2 years ago
- The Singular Values of Convolutional Layers☆71Updated 6 years ago
- Easy-to-use AdaHessian optimizer (PyTorch)☆77Updated 3 years ago