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:
- Meta-learning inductive biases in the form of useful conserved quantities.β37Updated 2 years ago
- A GPT, made only of MLPs, in Jaxβ58Updated 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
- β71Updated last year
- Python Research Frameworkβ106Updated 2 years ago
- Differentiable Algorithms and Algorithmic Supervision.β116Updated 2 years ago
- Usable implementation of Emerging Symbol Binding Network (ESBN), in Pytorchβ25Updated 4 years ago
- A collection of optimizers, some arcane others well known, for Flax.β29Updated 4 years ago
- Dive into Jax, Flax, XLA and C++β32Updated 5 years ago
- Code accompanying our paper "Feature Learning in Infinite-Width Neural Networks" (https://arxiv.org/abs/2011.14522)β63Updated 4 years ago
- β40Updated 2 years ago
- Toy implementations of some popular ML optimizers using Python/JAXβ44Updated 4 years ago
- β33Updated 4 years ago
- Framework-agnostic library for checking array/tensor shapes at runtime.β46Updated 4 years ago
- Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural netwoβ¦β72Updated 3 months ago
- β24Updated 6 years ago
- Official code repository of the paper Linear Transformers Are Secretly Fast Weight Programmers.β105Updated 4 years ago
- A lightweight library for tensorflow 2.0β66Updated 5 years ago
- a lightweight transformer library for PyTorchβ72Updated 3 years ago
- Implementation of deep implicit attention in PyTorchβ65Updated 4 years ago
- Experiments for Meta-Learning Symmetries by Reparameterizationβ57Updated 4 years ago
- Pytorch implementation of SuperPolyak subgradient method.β43Updated 2 years ago
- Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways - in Jax (Equinox framework)β189Updated 3 years ago
- Fast Discounted Cumulative Sums in PyTorchβ96Updated 4 years ago
- β115Updated last month
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).β112Updated 3 years ago
- β158Updated last year
- β106Updated last year
- [NeurIPS'19] Deep Equilibrium Models Jax Implementationβ40Updated 4 years ago
- β100Updated 3 years ago