goombalab / hydra
Official implementation of "Hydra: Bidirectional State Space Models Through Generalized Matrix Mixers"
☆119Updated 5 months ago
Alternatives and similar repositories for hydra:
Users that are interested in hydra are comparing it to the libraries listed below
- A Triton Kernel for incorporating Bi-Directionality in Mamba2☆60Updated last month
- Official PyTorch Implementation of "The Hidden Attention of Mamba Models"☆211Updated 8 months ago
- Reading list for research topics in state-space models☆257Updated last week
- Official implementation of Phi-Mamba. A MOHAWK-distilled model (Transformers to SSMs: Distilling Quadratic Knowledge to Subquadratic Mode…☆92Updated 4 months ago
- Awesome list of papers that extend Mamba to various applications.☆129Updated last month
- Implementation of Agent Attention in Pytorch☆89Updated 6 months ago
- An efficient pytorch implementation of selective scan in one file, works with both cpu and gpu, with corresponding mathematical derivatio…☆78Updated 10 months ago
- More dimensions = More fun☆21Updated 6 months ago
- Implementation of Infini-Transformer in Pytorch☆109Updated 3 weeks ago
- Implementation of a multimodal diffusion transformer in Pytorch☆99Updated 7 months ago
- PyTorch Implementation of Jamba: "Jamba: A Hybrid Transformer-Mamba Language Model"☆155Updated this week
- ☆243Updated 4 months ago
- Official PyTorch Implementation of Gated Delta Networks: Improving Mamba2 with Delta Rule☆112Updated 3 weeks ago
- ☆63Updated 3 months ago
- [NeurIPS 2024] Official implementation of the paper "MambaLRP: Explaining Selective State Space Sequence Models".☆36Updated 2 months ago
- Implementation of Soft MoE, proposed by Brain's Vision team, in Pytorch☆257Updated 9 months ago
- Simple, minimal implementation of the Mamba SSM in one pytorch file. Using logcumsumexp (Heisen sequence).☆106Updated 3 months ago
- ☆44Updated 2 months ago
- Some preliminary explorations of Mamba's context scaling.☆209Updated 11 months ago
- MambaFormer in-context learning experiments and implementation for https://arxiv.org/abs/2402.04248☆35Updated 7 months ago
- The this is the official implementation of "DAPE: Data-Adaptive Positional Encoding for Length Extrapolation"☆33Updated 3 months ago
- This repository is the official implementation of our Autoregressive Pretraining with Mamba in Vision☆68Updated 7 months ago
- [ECCV 2024] Official PyTorch implementation of RoPE-ViT "Rotary Position Embedding for Vision Transformer"☆270Updated last month
- [NeurIPS 2024] Official Repository of The Mamba in the Llama: Distilling and Accelerating Hybrid Models☆190Updated last week
- ☆45Updated 9 months ago
- Implementation of the proposed Adam-atan2 from Google Deepmind in Pytorch☆101Updated 2 months ago
- A repository for DenseSSMs☆86Updated 9 months ago
- Implementation of 🌻 Mirasol, SOTA Multimodal Autoregressive model out of Google Deepmind, in Pytorch☆88Updated last year
- Implementation of Griffin from the paper: "Griffin: Mixing Gated Linear Recurrences with Local Attention for Efficient Language Models"☆51Updated this week
- Official Implementation Of The Paper: `DeciMamba: Exploring the Length Extrapolation Potential of Mamba'☆22Updated 6 months ago