m-a-n-i-f-e-s-t / power-attentionLinks
Attention Kernels for Symmetric Power Transformers
☆128Updated 4 months ago
Alternatives and similar repositories for power-attention
Users that are interested in power-attention are comparing it to the libraries listed below
Sorting:
- MoE training for Me and You and maybe other people☆331Updated 3 weeks ago
- ☆109Updated 6 months ago
- DeMo: Decoupled Momentum Optimization☆198Updated last year
- Accelerated First Order Parallel Associative Scan☆196Updated 3 weeks ago
- seqax = sequence modeling + JAX☆170Updated 6 months ago
- A MAD laboratory to improve AI architecture designs 🧪☆135Updated last year
- supporting pytorch FSDP for optimizers☆84Updated last year
- Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clusters☆131Updated last year
- NanoGPT-speedrunning for the poor T4 enjoyers☆73Updated 9 months ago
- 📄Small Batch Size Training for Language Models☆80Updated 3 months ago
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆186Updated last week
- H-Net Dynamic Hierarchical Architecture☆81Updated 4 months ago
- Understand and test language model architectures on synthetic tasks.☆251Updated 2 weeks ago
- Normalized Transformer (nGPT)☆197Updated last year
- Supporting code for the blog post on modular manifolds.☆113Updated 4 months ago
- 🧱 Modula software package☆322Updated 5 months ago
- EvaByte: Efficient Byte-level Language Models at Scale☆115Updated 9 months ago
- ☆53Updated 2 years ago
- ☆70Updated last year
- PyTorch implementation of models from the Zamba2 series.☆186Updated last year
- Repo for "LoLCATs: On Low-Rank Linearizing of Large Language Models"☆251Updated last year
- an open source reproduction of NVIDIA's nGPT (Normalized Transformer with Representation Learning on the Hypersphere)☆109Updated 10 months ago
- Tiled Flash Linear Attention library for fast and efficient mLSTM Kernels.☆82Updated 2 months ago
- Dion optimizer algorithm☆420Updated 2 weeks ago
- Jax Codebase for Evolutionary Strategies at the Hyperscale☆216Updated last month
- Simple & Scalable Pretraining for Neural Architecture Research☆307Updated last month
- Custom triton kernels for training Karpathy's nanoGPT.☆19Updated last year
- ☆82Updated last year
- An implementation of PSGD Kron second-order optimizer for PyTorch☆98Updated 6 months ago
- The Automated LLM Speedrunning Benchmark measures how well LLM agents can reproduce previous innovations and discover new ones in languag…☆127Updated 3 months ago