Efficient Triton Kernels for LLM Training
-
Updated
Aug 3, 2026 - Python
Efficient Triton Kernels for LLM Training
FlagGems is an operator library for large language models implemented in the Triton Language.
LLM notes, including model inference, transformer model structure, and llm framework code analysis notes.
A light llama-like llm inference framework based on the triton kernel.
Tiled Flash Linear Attention library for fast and efficient mLSTM Kernels.
Production inference for encoder models - ColBERT, GLiNER, ColPali, embeddings etc. - as vLLM plugins for online and in-process deployment
Universal AI Runtime — Execute any model on any hardware
A "standard library" of Triton kernels.
High-performance late-interaction retrieval engine for on-prem AI. ColBERT/ColPali multi-vector search with Rust fused MaxSim, Triton GPU kernels, ROQ quantization, LEMUR routing, WAL-backed CRUD, and a FastAPI server — single machine, CPU or GPU.
Manifold-Constrained Hyper-Connections with fused Triton kernels for efficient training
Educational resource demonstrating common GPU programming pitfalls and solutions using Triton kernels.
Official Code for the paper ELMO : Efficiency via Low-precision and Peak Memory Optimization in Large Output Spaces (in ICML 2025)
A from-scratch nano-Triton and nano-Helion: the modern GPU-kernel DSL stack, rebuilt in ~4,000 lines
KernelHeim – development ground of custom Triton and CUDA kernel functions designed to optimize and accelerate machine learning workloads on NVIDIA GPUs. Inspired by the mythical stronghold of the gods, KernelHeim is a forge where high-performance kernels are crafted to unlock the full potential of the hardware.
LLM agents that generate, verify, and evolve Triton GPU kernels. Includes a reward-hack-resistant benchmarking harness with strict correctness verification and fresh-input evaluation. Achieves up to 174.7× over PyTorch eager and outperforms FlexAttention (1.48×) and SDPA (1.17×) on selected workloads.
A container of various PyTorch neural network modules written in Triton.
Better and Faster Implementation Dynamic Dilated Pyramid Module from HDFNet (ECCV 2020)
Research on implementing a custom Triton kernel for Fast-dLLM v2 and other Diffusion LLMs in order to produce model runtime speedups.
Reproducible benchmark suite and tuned Triton fused-MoE configs for NVIDIA H20 LLM inference. 24 configs, 36 perf data points, geomean 1.09× / peak 1.74× speedup.
ASR Pipeline (GLM-ASR) optimized using custom Triton kernels (achieving a 72.2% improvement in speed)
Add a description, image, and links to the triton-kernels topic page so that developers can more easily learn about it.
To associate your repository with the triton-kernels topic, visit your repo's landing page and select "manage topics."