jxbz / neroLinks
👑 Pytorch code for the Nero optimiser.
☆20Updated 2 years ago
Alternatives and similar repositories for nero
Users that are interested in nero are comparing it to the libraries listed below
Sorting:
- A GPT, made only of MLPs, in Jax☆58Updated 4 years ago
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago
- JAX implementation of Learning to learn by gradient descent by gradient descent☆27Updated 8 months ago
- Image augmentation library for Jax☆39Updated last year
- 👩 Pytorch and Jax code for the Madam optimiser.☆51Updated 4 years ago
- A collection of optimizers, some arcane others well known, for Flax.☆29Updated 3 years ago
- ☆33Updated 4 years ago
- AdaCat☆49Updated 2 years ago
- ☆68Updated last year
- A JAX implementation of stochastic addition.☆14Updated 2 years ago
- Toy implementations of some popular ML optimizers using Python/JAX☆44Updated 4 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 4 years ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆39Updated 4 years ago
- ☆80Updated 3 years ago
- ☆30Updated 3 years ago
- Usable implementation of Emerging Symbol Binding Network (ESBN), in Pytorch☆25Updated 4 years ago
- Framework-agnostic library for checking array/tensor shapes at runtime.☆46Updated 4 years ago
- Dive into Jax, Flax, XLA and C++☆31Updated 5 years ago
- Discontinuous Hamiltonian Monte Carlo in JAX☆41Updated 5 years ago
- Texture mapping with variational auto-encoders☆40Updated 3 years ago
- ☆43Updated 3 weeks ago
- 🧀 Pytorch code for the Fromage optimiser.☆124Updated 11 months ago
- ☆104Updated 11 months ago
- Very deep VAEs in JAX/Flax☆46Updated 4 years ago
- Experiments for Meta-Learning Symmetries by Reparameterization☆56Updated 4 years ago
- ☆24Updated 6 years ago
- ☆60Updated 3 years ago
- Official repository for our ICLR 2021 paper Evaluating the Disentanglement of Deep Generative Models with Manifold Topology☆36Updated 4 years ago
- Implementation of deep implicit attention in PyTorch☆65Updated 3 years ago
- Implementation of Token Shift GPT - An autoregressive model that solely relies on shifting the sequence space for mixing☆50Updated 3 years ago