jxbz / madam
👩 Pytorch and Jax code for the Madam optimiser.
☆51Updated 4 years ago
Alternatives and similar repositories for madam:
Users that are interested in madam are comparing it to the libraries listed below
- 🧀 Pytorch code for the Fromage optimiser.☆123Updated 8 months ago
- Structured matrices for compressing neural networks☆66Updated last year
- Experiment code for "Randomized Automatic Differentiation"☆66Updated 4 years ago
- ☆26Updated 5 years ago
- ☆61Updated last year
- ☆45Updated 5 years ago
- Autoregressive Energy Machines☆77Updated 2 years ago
- ☆153Updated 4 years ago
- Very deep VAEs in JAX/Flax☆46Updated 3 years ago
- Estimating Gradients for Discrete Random Variables by Sampling without Replacement☆40Updated 5 years ago
- ☆77Updated 4 years ago
- Padé Activation Units: End-to-end Learning of Activation Functions in Deep Neural Network☆64Updated 4 years ago
- Mixture Density Networks (Bishop, 1994) tutorial in JAX☆58Updated 4 years ago
- The Singular Values of Convolutional Layers☆72Updated 6 years ago
- A lightweight library for tensorflow 2.0☆66Updated 5 years ago
- This repository is no longer maintained. Check☆81Updated 4 years ago
- Code to accompany the paper Radial Bayesian Neural Networks: Beyond Discrete Support In Large-Scale Bayesian Deep Learning☆33Updated 4 years ago
- Statistical adaptive stochastic optimization methods☆33Updated 4 years ago
- [NeurIPS'19] [PyTorch] Adaptive Regularization in NN☆68Updated 5 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 4 years ago
- TBA☆75Updated 6 years ago
- Code for the Thermodynamic Variational Objective☆26Updated 2 years ago
- ☆12Updated 4 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆108Updated 2 years ago
- Normalizing Flows in Jax☆107Updated 4 years ago
- Computing the eigenvalues of Neural Tangent Kernel and Conjugate Kernel (aka NNGP kernel) over the boolean cube☆48Updated 5 years ago
- ☆99Updated 3 years ago
- Easy-to-use AdaHessian optimizer (PyTorch)☆77Updated 4 years ago
- Framework-agnostic library for checking array/tensor shapes at runtime.☆47Updated 3 years ago
- A library for evaluating representations.☆76Updated 3 years ago