AI9Stars / AutoTritonLinks
☆48Updated 2 months ago
Alternatives and similar repositories for AutoTriton
Users that are interested in AutoTriton are comparing it to the libraries listed below
Sorting:
- The evaluation framework for training-free sparse attention in LLMs☆101Updated 3 months ago
- The official implementation of paper: SimLayerKV: A Simple Framework for Layer-Level KV Cache Reduction.☆49Updated 11 months ago
- The source code of our work "Prepacking: A Simple Method for Fast Prefilling and Increased Throughput in Large Language Models" [AISTATS …☆60Updated last year
- ☆143Updated 7 months ago
- Long Context Extension and Generalization in LLMs☆60Updated last year
- ☆129Updated 4 months ago
- Official implementation for DenseMixer: Improving MoE Post-Training with Precise Router Gradient☆58Updated 2 months ago
- ☆119Updated 4 months ago
- 🔥 LLM-powered GPU kernel synthesis: Train models to convert PyTorch ops into optimized Triton kernels via SFT+RL. Multi-turn compilation…☆79Updated last week
- An efficient implementation of the NSA (Native Sparse Attention) kernel☆119Updated 3 months ago
- The official repository for SkyLadder: Better and Faster Pretraining via Context Window Scheduling☆35Updated last month
- Work in progress.☆74Updated 3 months ago
- ☆54Updated 4 months ago
- Flash-Muon: An Efficient Implementation of Muon Optimizer☆193Updated 3 months ago
- ☆96Updated last month
- A scalable asynchronous reinforcement learning implementation with in-flight weight updates.☆207Updated this week
- The simplest implementation of recent Sparse Attention patterns for efficient LLM inference.☆89Updated 2 months ago
- Implementation for FP8/INT8 Rollout for RL training without performence drop.☆253Updated last week
- [ICLR2025] Breaking Throughput-Latency Trade-off for Long Sequences with Speculative Decoding☆130Updated 10 months ago
- ☆14Updated last year
- ☆61Updated 3 months ago
- [COLM 2025] Code for Paper: Learning Adaptive Parallel Reasoning with Language Models☆130Updated last month
- Official implementation for Training LLMs with MXFP4☆96Updated 5 months ago
- Make reasoning models scalable☆47Updated 4 months ago
- Ouroboros: Speculative Decoding with Large Model Enhanced Drafting (EMNLP 2024 main)☆110Updated 6 months ago
- ☆70Updated 2 months ago
- ☆106Updated last year
- ☆52Updated 11 months ago
- Memory optimized Mixture of Experts☆68Updated 2 months ago
- All information and news with respect to Falcon-H1 series☆90Updated this week