jxbz / madamLinks
👩 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
Sorting:
- 🧀 Pytorch code for the Fromage optimiser.☆124Updated 10 months ago
- Mixture Density Networks (Bishop, 1994) tutorial in JAX☆59Updated 5 years ago
- ☆99Updated 3 years ago
- Experiment code for "Randomized Automatic Differentiation"☆67Updated 4 years ago
- Structured matrices for compressing neural networks☆66Updated last year
- ☆45Updated 5 years ago
- ☆153Updated 5 years ago
- Very deep VAEs in JAX/Flax☆46Updated 3 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 4 years ago
- A lightweight library for tensorflow 2.0☆66Updated 5 years ago
- ☆67Updated last year
- Codebase for Learning Invariances in Neural Networks☆95Updated 2 years ago
- ☆49Updated 4 years ago
- Autoregressive Energy Machines☆78Updated 2 years ago
- A minimal implementation of a VAE with BinConcrete (relaxed Bernoulli) latent distribution in TensorFlow.☆22Updated 5 years ago
- Padé Activation Units: End-to-end Learning of Activation Functions in Deep Neural Network☆64Updated 4 years ago
- ☆12Updated 4 years ago
- Experiments for Meta-Learning Symmetries by Reparameterization☆56Updated 4 years ago
- Pretrained TorchVision models on CIFAR10 dataset (with weights)☆24Updated 4 years ago
- Easy-to-use AdaHessian optimizer (PyTorch)☆78Updated 4 years ago
- Odds and Ends and Things I've implemented.☆78Updated 6 years ago
- Code to accompany the paper Radial Bayesian Neural Networks: Beyond Discrete Support In Large-Scale Bayesian Deep Learning☆33Updated 5 years ago
- ☆61Updated 2 years ago
- ☆80Updated 3 years ago
- Estimating Gradients for Discrete Random Variables by Sampling without Replacement☆40Updated 5 years ago
- ☆32Updated 6 years ago
- A discrete sequential VAE☆39Updated 5 years ago
- Limitations of the Empirical Fisher Approximation☆47Updated 3 months ago
- Framework-agnostic library for checking array/tensor shapes at runtime.☆46Updated 4 years ago
- A JAX implementation of stochastic addition.☆14Updated 2 years ago