jxbz / neroLinks
π Pytorch code for the Nero optimiser.
β20Updated 3 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.β38Updated 3 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"β60Updated 3 years ago
- β24Updated 6 years ago
- Differentiable Algorithms and Algorithmic Supervision.β116Updated 2 years ago
- Code accompanying our paper "Feature Learning in Infinite-Width Neural Networks" (https://arxiv.org/abs/2011.14522)β63Updated 4 years ago
- A collection of optimizers, some arcane others well known, for Flax.β29Updated 4 years ago
- Implementation of deep implicit attention in PyTorchβ65Updated 4 years ago
- β71Updated 2 years ago
- Dive into Jax, Flax, XLA and C++β32Updated 5 years ago
- Usable implementation of Emerging Symbol Binding Network (ESBN), in Pytorchβ25Updated 4 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.β45Updated 4 months ago
- Python Research Frameworkβ106Updated 3 years ago
- β107Updated last year
- β161Updated last year
- β40Updated 3 years ago
- β108Updated 2 years ago
- Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural netwoβ¦β74Updated 5 months ago
- β118Updated last month
- A lightweight library for tensorflow 2.0β65Updated 6 years ago
- JAX implementation of "Fine-Tuning Language Models with Just Forward Passes"β19Updated 2 years ago
- A case study of efficient training of large language models using commodity hardware.β68Updated 3 years ago
- A framework for implementing equivariant DLβ10Updated 4 years ago
- Documentation:β125Updated 2 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).β117Updated 3 years ago
- Efficient Householder Transformation in PyTorchβ67Updated 4 years ago
- Official code repository of the paper Linear Transformers Are Secretly Fast Weight Programmers.β110Updated 4 years ago
- JMP is a Mixed Precision library for JAX.β211Updated 10 months ago
- A small library for creating and manipulating custom JAX Pytree classesβ56Updated 2 years ago
- Structured matrices for compressing neural networksβ67Updated 2 years ago