acosharma / elita-transformerLinks
Official Repository for Efficient Linear-Time Attention Transformers.
☆18Updated last year
Alternatives and similar repositories for elita-transformer
Users that are interested in elita-transformer are comparing it to the libraries listed below
Sorting:
- ☆32Updated last year
- Efficient PScan implementation in PyTorch☆16Updated last year
- sigma-MoE layer☆18Updated last year
- Fine-Tuning Pre-trained Transformers into Decaying Fast Weights☆19Updated 2 years ago
- Parallel Associative Scan for Language Models☆18Updated last year
- Blog post☆17Updated last year
- ☆20Updated last year
- My explorations into editing the knowledge and memories of an attention network☆35Updated 2 years ago
- [NeurIPS 2023] Sparse Modular Activation for Efficient Sequence Modeling☆36Updated last year
- Engineering the state of RNN language models (Mamba, RWKV, etc.)☆32Updated last year
- ☆29Updated 2 years ago
- ☆17Updated last month
- Official Code Repository for the paper "Key-value memory in the brain"☆26Updated 3 months ago
- Reference implementation of "Softmax Attention with Constant Cost per Token" (Heinsen, 2024)☆24Updated 11 months ago
- Combining SOAP and MUON☆16Updated 3 months ago
- Official repository of paper "RNNs Are Not Transformers (Yet): The Key Bottleneck on In-context Retrieval"☆27Updated last year
- ☆11Updated last year
- BigKnow2022: Bringing Language Models Up to Speed☆15Updated 2 years ago
- ☆19Updated 2 weeks ago
- Official repository for the paper "Approximating Two-Layer Feedforward Networks for Efficient Transformers"☆37Updated last year
- ☆46Updated last year
- Implementation of Token Shift GPT - An autoregressive model that solely relies on shifting the sequence space for mixing☆50Updated 3 years ago
- AdamW optimizer for bfloat16 models in pytorch 🔥.☆32Updated 11 months ago
- RWKV model implementation☆38Updated last year
- Experiments on the impact of depth in transformers and SSMs.☆30Updated 7 months ago
- JAX Scalify: end-to-end scaled arithmetics☆16Updated 7 months ago
- Triton Implementation of HyperAttention Algorithm☆48Updated last year
- Expanding linear RNN state-transition matrix eigenvalues to include negatives improves state-tracking tasks and language modeling without…☆15Updated 2 months ago
- Code for the note "NF4 Isn't Information Theoretically Optimal (and that's Good)☆19Updated last year
- ☆14Updated 2 years ago