Edward-Sun / gpt-accelera
Simple and efficient pytorch-native transformer training and inference (batched)
☆68Updated 10 months ago
Alternatives and similar repositories for gpt-accelera:
Users that are interested in gpt-accelera are comparing it to the libraries listed below
- PyTorch building blocks for the OLMo ecosystem☆54Updated this week
- Language models scale reliably with over-training and on downstream tasks☆96Updated 10 months ago
- The simplest implementation of recent Sparse Attention patterns for efficient LLM inference.☆57Updated 3 weeks ago
- The source code of our work "Prepacking: A Simple Method for Fast Prefilling and Increased Throughput in Large Language Models"☆59Updated 4 months ago
- A framework for few-shot evaluation of autoregressive language models.☆24Updated last year
- ☆82Updated 4 months ago
- A toolkit for scaling law research ⚖☆47Updated 3 weeks ago
- Code for NeurIPS 2024 Spotlight: "Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durations"☆70Updated 3 months ago
- ☆48Updated last year
- ☆51Updated 9 months ago
- ☆58Updated 9 months ago
- Official repository of paper "RNNs Are Not Transformers (Yet): The Key Bottleneck on In-context Retrieval"☆25Updated 10 months ago
- Long Context Extension and Generalization in LLMs☆48Updated 4 months ago
- Code and Configs for Asynchronous RLHF: Faster and More Efficient RL for Language Models☆31Updated 2 months ago
- Replicating O1 inference-time scaling laws☆82Updated 2 months ago
- ☆49Updated 7 months ago
- ☆75Updated 7 months ago
- Stick-breaking attention☆43Updated last month
- NAACL '24 (Best Demo Paper RunnerUp) / MlSys @ NeurIPS '23 - RedCoast: A Lightweight Tool to Automate Distributed Training and Inference☆64Updated 2 months ago
- ☆47Updated last year
- ☆80Updated 11 months ago
- ☆91Updated 8 months ago
- Official github repo for the paper "Compression Represents Intelligence Linearly" [COLM 2024]☆130Updated 5 months ago
- Linear Attention Sequence Parallelism (LASP)☆77Updated 8 months ago
- NeurIPS 2024 tutorial on LLM Inference☆39Updated 2 months ago
- ☆30Updated 11 months ago
- Using FlexAttention to compute attention with different masking patterns☆40Updated 4 months ago
- A fusion of a linear layer and a cross entropy loss, written for pytorch in triton.☆62Updated 6 months ago
- ☆71Updated 6 months ago