CLAIRE-Labo / StructuredFFNLinks
The official code of "Building on Efficient Foundations: Effectively Training LLMs with Structured Feedforward Layers"
☆19Updated last year
Alternatives and similar repositories for StructuredFFN
Users that are interested in StructuredFFN are comparing it to the libraries listed below
Sorting:
- ☆53Updated last year
- Tiled Flash Linear Attention library for fast and efficient mLSTM Kernels.☆79Updated last month
- ☆51Updated last year
- Accelerated First Order Parallel Associative Scan☆193Updated last year
- Supporting code for the blog post on modular manifolds.☆107Updated 2 months ago
- Griffin MQA + Hawk Linear RNN Hybrid☆89Updated last year
- ☆44Updated 7 years ago
- ☆62Updated last year
- An implementation of the Llama architecture, to instruct and delight☆21Updated 6 months ago
- Muon fsdp 2☆46Updated 4 months ago
- Efficient PScan implementation in PyTorch☆17Updated last year
- ☆50Updated last week
- ☆121Updated last year
- A toolkit for scaling law research ⚖☆53Updated 10 months ago
- Combining SOAP and MUON☆17Updated 10 months ago
- Triton Implementation of HyperAttention Algorithm☆48Updated 2 years ago
- ☆91Updated last year
- Official repository for the paper "Approximating Two-Layer Feedforward Networks for Efficient Transformers"☆38Updated 6 months ago
- A MAD laboratory to improve AI architecture designs 🧪☆135Updated last year
- Code for exploring Based models from "Simple linear attention language models balance the recall-throughput tradeoff"☆243Updated 6 months ago
- Implementation of GateLoop Transformer in Pytorch and Jax☆91Updated last year
- A fusion of a linear layer and a cross entropy loss, written for pytorch in triton.☆74Updated last year
- ☆19Updated 3 weeks ago
- Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training☆132Updated last year
- Parallel Associative Scan for Language Models☆18Updated last year
- A library for unit scaling in PyTorch☆133Updated 5 months ago
- DPO, but faster 🚀☆46Updated last year
- supporting pytorch FSDP for optimizers☆84Updated last year
- Experiments on the impact of depth in transformers and SSMs.☆39Updated 2 months ago
- [NeurIPS 2023 spotlight] Official implementation of HGRN in our NeurIPS 2023 paper - Hierarchically Gated Recurrent Neural Network for Se…☆66Updated last year