meta-pytorch / KernelAgentLinks
Autonomous GPU Kernel Generation via Deep Agents
☆228Updated this week
Alternatives and similar repositories for KernelAgent
Users that are interested in KernelAgent are comparing it to the libraries listed below
Sorting:
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆324Updated this week
- PyTorch bindings for CUTLASS grouped GEMM.☆142Updated 8 months ago
- Accelerating MoE with IO and Tile-aware Optimizations☆569Updated 2 weeks ago
- Building the Virtuous Cycle for AI-driven LLM Systems☆151Updated this week
- ☆104Updated last year
- ☆159Updated last year
- QuTLASS: CUTLASS-Powered Quantized BLAS for Deep Learning☆165Updated 2 months ago
- TritonBench: Benchmarking Large Language Model Capabilities for Generating Triton Operators☆114Updated 7 months ago
- FlashInfer Bench @ MLSys 2026: Building AI agents to write high performance GPU kernels☆84Updated 2 weeks ago
- DeeperGEMM: crazy optimized version☆73Updated 9 months ago
- Distributed MoE in a Single Kernel [NeurIPS '25]☆190Updated last week
- Code for the paper “Four Over Six: More Accurate NVFP4 Quantization with Adaptive Block Scaling”☆118Updated this week
- ☆65Updated 9 months ago
- ☆259Updated last year
- A lightweight design for computation-communication overlap.☆219Updated 2 weeks ago
- extensible collectives library in triton☆95Updated 10 months ago
- ☆117Updated 8 months ago
- An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).☆276Updated 6 months ago
- Collection of kernels written in Triton language☆178Updated last week
- ☆35Updated 11 months ago
- Framework to reduce autotune overhead to zero for well known deployments.☆95Updated 4 months ago
- Cataloging released Triton kernels.☆292Updated 4 months ago
- ☆85Updated last year
- Implement Flash Attention using Cute.☆100Updated last year
- Flash-LLM: Enabling Cost-Effective and Highly-Efficient Large Generative Model Inference with Unstructured Sparsity☆233Updated 2 years ago
- Code for paper: [ICLR2025 Oral] FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inference☆160Updated 3 months ago
- ☆286Updated this week
- nnScaler: Compiling DNN models for Parallel Training☆124Updated 4 months ago
- 16-fold memory access reduction with nearly no loss☆110Updated 10 months ago
- NVIDIA cuTile learn☆158Updated last month