lucidrains / minGRU-pytorchLinks
Implementation of the proposed minGRU in Pytorch
☆300Updated 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☆115Updated 8 months ago
- Attempt to make multiple residual streams from Bytedance's Hyper-Connections paper accessible to the public☆88Updated last month
- Pytorch implementation of the xLSTM model by Beck et al. (2024)☆169Updated 11 months ago
- Pytorch implementation of Simplified Structured State-Spaces for Sequence Modeling (S5)☆77Updated last year
- The AdEMAMix Optimizer: Better, Faster, Older.☆184Updated 10 months ago
- A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch☆100Updated last year
- PyTorch implementation of Structured State Space for Sequence Modeling (S4), based on Annotated S4.☆83Updated last year
- A State-Space Model with Rational Transfer Function Representation.☆79Updated last year
- ☆298Updated 7 months ago
- Quick implementation of nGPT, learning entirely on the hypersphere, from NvidiaAI☆289Updated 2 months ago
- Simple, minimal implementation of the Mamba SSM in one pytorch file. Using logcumsumexp (Heisen sequence).☆120Updated 9 months ago
- Implementation of GateLoop Transformer in Pytorch and Jax☆89Updated last year
- When it comes to optimizers, it's always better to be safe than sorry☆352Updated this week
- ☆206Updated 8 months ago
- Explorations into whether a transformer with RL can direct a genetic algorithm to converge faster☆70Updated 2 months ago
- Implementation of Griffin from the paper: "Griffin: Mixing Gated Linear Recurrences with Local Attention for Efficient Language Models"☆56Updated 2 weeks ago
- Implementation of Liquid Nets in Pytorch☆67Updated 2 weeks ago
- Implementation of Agent Attention in Pytorch☆91Updated last year
- Implementation of Block Recurrent Transformer - Pytorch☆221Updated 11 months ago
- Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new…☆124Updated last year
- Oscillatory State-Space Models☆96Updated 4 months ago
- Kolmogorov-Arnold Networks (KAN) using Chebyshev polynomials instead of B-splines.☆383Updated last year
- ☆163Updated 2 years ago
- Pytorch (Lightning) implementation of the Mamba model☆29Updated 3 months ago
- Minimal Mamba-2 implementation in PyTorch☆212Updated last year
- Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"☆101Updated 7 months ago
- Library for Jacobian descent with PyTorch. It enables the optimization of neural networks with multiple losses (e.g. multi-task learning)…☆264Updated this week
- A simple but robust PyTorch implementation of RetNet from "Retentive Network: A Successor to Transformer for Large Language Models" (http…☆106Updated last year
- Reading list for research topics in state-space models☆315Updated last month
- Reinforcement Learning example in Nim, playing tic tac toe. Based off original C version from the great Antirez☆14Updated 4 months ago