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
- Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"β59Updated 3 years ago
- Differentiable Algorithms and Algorithmic Supervision.β115Updated 2 years ago
- β68Updated last year
- A GPT, made only of MLPs, in Jaxβ58Updated 4 years ago
- β104Updated last year
- Code accompanying our paper "Feature Learning in Infinite-Width Neural Networks" (https://arxiv.org/abs/2011.14522)β62Updated 4 years ago
- Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways - in Jax (Equinox framework)β187Updated 3 years ago
- Fast Discounted Cumulative Sums in PyTorchβ96Updated 3 years ago
- Usable implementation of Emerging Symbol Binding Network (ESBN), in Pytorchβ25Updated 4 years ago
- Python Research Frameworkβ106Updated 2 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).β112Updated 3 years ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementationβ40Updated 4 years ago
- β40Updated last year
- β158Updated last year
- β24Updated 6 years ago
- JMP is a Mixed Precision library for JAX.β207Updated 6 months ago
- β115Updated this week
- minGPT in JAXβ48Updated 3 years ago
- β108Updated 2 years ago
- β39Updated 2 years ago
- A stateful pytree library for training neural networks.β22Updated 3 years ago
- Framework-agnostic library for checking array/tensor shapes at runtime.β46Updated 4 years ago
- Lightweight Cluster/Cloud VM Job Management πβ42Updated 11 months ago
- A small library for creating and manipulating custom JAX Pytree classesβ56Updated 2 years ago
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation preconditionβ¦β180Updated last week
- β33Updated 4 years ago
- β60Updated 3 years ago
- Easy-to-use AdaHessian optimizer (PyTorch)β79Updated 4 years ago
- A functional training loops library for JAXβ88Updated last year