lucidrains / minGRU-pytorchLinks
Implementation of the proposed minGRU in Pytorch
☆301Updated 4 months ago
Alternatives and similar repositories for minGRU-pytorch
Users that are interested in minGRU-pytorch are comparing it to the libraries listed below
Sorting:
- Implementation of the proposed Adam-atan2 from Google Deepmind in Pytorch☆110Updated 7 months ago
- ☆197Updated 7 months ago
- Attempt to make multiple residual streams from Bytedance's Hyper-Connections paper accessible to the public☆87Updated 3 weeks ago
- Pytorch implementation of the xLSTM model by Beck et al. (2024)☆168Updated 11 months ago
- Simple, minimal implementation of the Mamba SSM in one pytorch file. Using logcumsumexp (Heisen sequence).☆120Updated 8 months ago
- The AdEMAMix Optimizer: Better, Faster, Older.☆183Updated 10 months ago
- A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch☆99Updated last year
- Pytorch implementation of Simplified Structured State-Spaces for Sequence Modeling (S5)☆77Updated last year
- Library for Jacobian descent with PyTorch. It enables the optimization of neural networks with multiple losses (e.g. multi-task learning)…☆251Updated this week
- A State-Space Model with Rational Transfer Function Representation.☆79Updated last year
- Integrating Mamba/SSMs with Transformer for Enhanced Long Context and High-Quality Sequence Modeling☆197Updated 3 months ago
- When it comes to optimizers, it's always better to be safe than sorry☆246Updated 3 months ago
- Code for Adam-mini: Use Fewer Learning Rates To Gain More https://arxiv.org/abs/2406.16793☆427Updated 2 months ago
- Quick implementation of nGPT, learning entirely on the hypersphere, from NvidiaAI☆287Updated last month
- Implementation of Rotary Embeddings, from the Roformer paper, in Pytorch☆708Updated last week
- Training small GPT-2 style models using Kolmogorov-Arnold networks.☆120Updated last year
- Implementation of Griffin from the paper: "Griffin: Mixing Gated Linear Recurrences with Local Attention for Efficient Language Models"☆55Updated 3 months ago
- Explorations into whether a transformer with RL can direct a genetic algorithm to converge faster☆70Updated last month
- ☆295Updated 6 months ago
- Kolmogorov-Arnold Networks (KAN) using Chebyshev polynomials instead of B-splines.☆380Updated last year
- ☆121Updated last month
- Implementation of Block Recurrent Transformer - Pytorch☆220Updated 10 months ago
- Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new…☆123Updated 11 months ago
- Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"☆101Updated 6 months ago
- ☆292Updated 7 months ago
- PyTorch implementation of Structured State Space for Sequence Modeling (S4), based on Annotated S4.☆82Updated last year
- Official Implementation of "ADOPT: Modified Adam Can Converge with Any β2 with the Optimal Rate"☆426Updated 7 months ago
- ☆290Updated 2 months ago
- A simple but robust PyTorch implementation of RetNet from "Retentive Network: A Successor to Transformer for Large Language Models" (http…☆106Updated last year
- Implementation of Agent Attention in Pytorch☆90Updated last year