🔥 LLM-powered GPU kernel synthesis: Train models to convert PyTorch ops into optimized Triton kernels via SFT+RL. Multi-turn compilation feedback, cross-platform NVIDIA/AMD, Kernelbook + KernelBench
☆132Nov 10, 2025Updated 4 months ago
Alternatives and similar repositories for TritonForge
Users that are interested in TritonForge are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- APRIL: Active Partial Rollouts in Reinforcement Learning to Tame Long-tail Generation. A system-level optimization for scalable LLM tra…☆54Oct 11, 2025Updated 5 months ago
- ☆28Jul 29, 2025Updated 7 months ago
- ☆76Sep 15, 2025Updated 6 months ago
- [NeurIPS 2025] Scaling Speculative Decoding with Lookahead Reasoning☆67Oct 31, 2025Updated 4 months ago
- [ICLR 2025] DeFT: Decoding with Flash Tree-attention for Efficient Tree-structured LLM Inference☆50Jun 17, 2025Updated 9 months ago
- ☆20Jun 3, 2023Updated 2 years ago
- OpenAI gym environments for goal-conditioned and language-conditioned reinforcement learning☆14Jan 27, 2026Updated last month
- An agent for CUDA compute-communication kernel co-design☆32Mar 11, 2026Updated last week
- Estimate MFU for DeepSeekV3☆26Jan 5, 2025Updated last year
- Bridge Megatron-Core to Hugging Face/Reinforcement Learning☆204Updated this week
- Implementation for FP8/INT8 Rollout for RL training without performence drop.☆297Nov 7, 2025Updated 4 months ago
- [ASPLOS'26] Taming the Long-Tail: Efficient Reasoning RL Training with Adaptive Drafter☆156Feb 27, 2026Updated 3 weeks ago
- Moatless Testbeds allows you to create isolated testbed environments in a Kubernetes cluster where you can apply code changes through git…☆14Apr 9, 2025Updated 11 months ago
- Quantized Attention on GPU☆44Nov 22, 2024Updated last year
- Collection of kernels written in Triton language☆184Jan 27, 2026Updated last month
- ☆36Feb 12, 2025Updated last year
- slime is an LLM post-training framework for RL Scaling.☆4,906Updated this week
- PaperHelper: Knowledge-Based LLM QA Paper Reading Assistant with Reliable References☆21Jun 13, 2024Updated last year
- ☆15Jul 13, 2025Updated 8 months ago
- ☆52May 19, 2025Updated 10 months ago
- [DEPRECATED] Moved to ROCm/rocm-libraries repo☆139Mar 13, 2026Updated last week
- [ICML 2025] SparseLoRA: Accelerating LLM Fine-Tuning with Contextual Sparsity☆71Mar 10, 2026Updated last week
- Materials for learning SGLang☆775Jan 5, 2026Updated 2 months ago
- [ICLR 2025] COAT: Compressing Optimizer States and Activation for Memory-Efficient FP8 Training☆262Aug 9, 2025Updated 7 months ago
- Utility scripts for PyTorch (e.g. Make Perfetto show some disappearing kernels, Memory profiler that understands more low-level allocatio…☆93Sep 11, 2025Updated 6 months ago
- AMD RAD's multi-GPU Triton-based framework for seamless multi-GPU programming☆182Updated this week
- A NCCL extension library, designed to efficiently offload GPU memory allocated by the NCCL communication library.☆100Dec 17, 2025Updated 3 months ago
- ☆22May 5, 2025Updated 10 months ago
- Research prototype of PRISM — a cost-efficient multi-LLM serving system with flexible time- and space-based GPU sharing.☆58Mar 17, 2026Updated last week
- Measuring the Signal to Noise Ratio in Language Model Evaluation☆29Aug 19, 2025Updated 7 months ago
- Fast, memory-efficient attention column reduction (e.g., sum, mean, max)☆42Feb 10, 2026Updated last month
- Drop-in environment replacements that make your RL algorithm train faster.☆21Jun 19, 2024Updated last year
- Implementation from scratch in C of the Multi-head latent attention used in the Deepseek-v3 technical paper.☆18Jan 15, 2025Updated last year
- [ICML2025] SpargeAttention: A training-free sparse attention that accelerates any model inference.☆961Feb 25, 2026Updated 3 weeks ago
- ☆15Aug 19, 2025Updated 7 months ago
- ☆10Mar 13, 2026Updated last week
- Wave: Python Domain-Specific Language for High Performance Machine Learning☆48Updated this week
- [NeurIPS 2025] Simple extension on vLLM to help you speed up reasoning model without training.☆224May 31, 2025Updated 9 months ago
- A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.☆809Updated this week