CLAIRE-Labo / StructuredFFN
The official code of "Building on Efficient Foundations: Effectively Training LLMs with Structured Feedforward Layers"
☆17Updated 7 months ago
Alternatives and similar repositories for StructuredFFN:
Users that are interested in StructuredFFN are comparing it to the libraries listed below
- Official repository for the paper "Approximating Two-Layer Feedforward Networks for Efficient Transformers"☆36Updated last year
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.☆30Updated 2 months ago
- ☆19Updated 5 months ago
- ☆52Updated 5 months ago
- An implementation of the Llama architecture, to instruct and delight☆21Updated last month
- RWKV model implementation☆37Updated last year
- Code Release for "Broken Neural Scaling Laws" (BNSL) paper☆58Updated last year
- Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.☆17Updated this week
- Triton Implementation of HyperAttention Algorithm☆47Updated last year
- ☆47Updated last year
- GoldFinch and other hybrid transformer components☆45Updated 7 months ago
- ☆30Updated 3 months ago
- Machine Learning eXperiment Utilities☆46Updated 9 months ago
- ☆38Updated last year
- Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch☆23Updated last month
- Parallel Associative Scan for Language Models☆18Updated last year
- Mixture of A Million Experts☆42Updated 7 months ago
- Official Code Repository for the paper "Key-value memory in the brain"☆23Updated 2 weeks ago
- ☆27Updated last year
- [ICLR 2025] "Training LMs on Synthetic Edit Sequences Improves Code Synthesis" (Piterbarg, Pinto, Fergus)☆15Updated last month
- Codes accompanying the paper "LaProp: a Better Way to Combine Momentum with Adaptive Gradient"☆27Updated 4 years ago
- ☆18Updated 9 months ago
- ☆57Updated 3 months ago
- Exploration into the proposed "Self Reasoning Tokens" by Felipe Bonetto☆55Updated 9 months ago
- ☆30Updated last year
- Learn online intrinsic rewards from LLM feedback☆34Updated 2 months ago
- Engineering the state of RNN language models (Mamba, RWKV, etc.)☆32Updated 9 months ago
- Experiments on the impact of depth in transformers and SSMs.☆23Updated 4 months ago