m-a-n-i-f-e-s-t / power-attentionLinks
Attention Kernels for Symmetric Power Transformers
β128Updated 2 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:
- A MAD laboratory to improve AI architecture designs π§ͺβ135Updated 11 months ago
- DeMo: Decoupled Momentum Optimizationβ197Updated 11 months ago
- β105Updated 4 months ago
- H-Net Dynamic Hierarchical Architectureβ80Updated 2 months ago
- Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clustersβ130Updated 11 months ago
- The simplest, fastest repository for training/finetuning medium-sized GPTs.β173Updated 5 months ago
- seqax = sequence modeling + JAXβ168Updated 4 months ago
- Accelerated First Order Parallel Associative Scanβ192Updated last year
- supporting pytorch FSDP for optimizersβ84Updated 11 months ago
- β68Updated last year
- PyTorch implementation of models from the Zamba2 series.β185Updated 10 months ago
- Understand and test language model architectures on synthetic tasks.β240Updated 2 months ago
- Normalized Transformer (nGPT)β194Updated last year
- EvaByte: Efficient Byte-level Language Models at Scaleβ111Updated 7 months ago
- an open source reproduction of NVIDIA's nGPT (Normalized Transformer with Representation Learning on the Hypersphere)β108Updated 8 months ago
- Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT trainingβ132Updated last year
- πSmall Batch Size Training for Language Modelsβ63Updated last month
- Repo for "LoLCATs: On Low-Rank Linearizing of Large Language Models"β249Updated 10 months ago
- Simple & Scalable Pretraining for Neural Architecture Researchβ302Updated last month
- β53Updated last year
- π§± Modula software packageβ307Updated 3 months ago
- Supporting code for the blog post on modular manifolds.β103Updated 2 months ago
- β201Updated 3 months ago
- The Automated LLM Speedrunning Benchmark measures how well LLM agents can reproduce previous innovations and discover new ones in languagβ¦β112Updated last month
- Tiled Flash Linear Attention library for fast and efficient mLSTM Kernels.β74Updated 2 weeks ago
- Code for exploring Based models from "Simple linear attention language models balance the recall-throughput tradeoff"β243Updated 5 months ago
- β91Updated last year
- A set of Python scripts that makes your experience on TPU betterβ54Updated 2 months ago
- β82Updated last year
- Minimal yet performant LLM examples in pure JAXβ202Updated 2 months ago