jxbz / nero
π 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
- Meta-learning inductive biases in the form of useful conserved quantities.β37Updated 2 years ago
- A GPT, made only of MLPs, in Jaxβ57Updated 3 years ago
- π© Pytorch and Jax code for the Madam optimiser.β51Updated 4 years ago
- β33Updated 4 years ago
- Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"β59Updated 3 years ago
- Usable implementation of Emerging Symbol Binding Network (ESBN), in Pytorchβ24Updated 4 years ago
- Official repository for our ICLR 2021 paper Evaluating the Disentanglement of Deep Generative Models with Manifold Topologyβ36Updated 4 years ago
- JAX implementation of Learning to learn by gradient descent by gradient descentβ27Updated 5 months ago
- β30Updated 3 years ago
- β68Updated last year
- A collection of optimizers, some arcane others well known, for Flax.β29Updated 3 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.β44Updated 4 years ago
- A framework for implementing equivariant DLβ10Updated 3 years ago
- A python library for highly configurable transformers - easing model architecture search and experimentation.β49Updated 3 years ago
- β24Updated 6 years ago
- Very deep VAEs in JAX/Flaxβ46Updated 3 years ago
- A JAX implementation of stochastic addition.β14Updated 2 years ago
- A simple Transformer where the softmax has been replaced with normalizationβ19Updated 4 years ago
- Discontinuous Hamiltonian Monte Carlo in JAXβ41Updated 5 years ago
- code for "Semi-Discrete Normalizing Flows through Differentiable Tessellation"β26Updated 2 years ago
- Toy implementations of some popular ML optimizers using Python/JAXβ44Updated 3 years ago
- AdaCatβ49Updated 2 years ago
- β21Updated 2 years ago
- High performance pytorch modulesβ18Updated 2 years ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementationβ39Updated 4 years ago
- Image augmentation library for Jaxβ39Updated 11 months ago
- Inference on non-linear dynamical systems written in JAXβ11Updated 4 years ago
- A collection of Models, Datasets, DataModules, Callbacks, Metrics, Losses and Loggers to better integrate pytorch-lightning with transforβ¦β47Updated last year
- Implementation of Token Shift GPT - An autoregressive model that solely relies on shifting the sequence space for mixingβ48Updated 3 years ago
- Implementation of deep implicit attention in PyTorchβ65Updated 3 years ago