meta-pytorch / KernelAgentLinks
Autonomous GPU Kernel Generation via Deep Agents
☆202Updated this week
Alternatives and similar repositories for KernelAgent
Users that are interested in KernelAgent are comparing it to the libraries listed below
Sorting:
- PyTorch bindings for CUTLASS grouped GEMM.☆139Updated 7 months ago
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆308Updated this week
- ☆100Updated last year
- Accelerating MoE with IO and Tile-aware Optimizations☆522Updated this week
- DeeperGEMM: crazy optimized version☆74Updated 8 months ago
- ☆65Updated 8 months ago
- QuTLASS: CUTLASS-Powered Quantized BLAS for Deep Learning☆155Updated last month
- Building the Virtuous Cycle for AI-driven LLM Systems☆112Updated this week
- ☆153Updated last year
- extensible collectives library in triton☆91Updated 9 months ago
- TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.☆104Updated 6 months ago
- ☆116Updated 7 months ago
- Implement Flash Attention using Cute.☆100Updated last year
- A lightweight design for computation-communication overlap.☆207Updated 2 weeks ago
- Code for the paper “Four Over Six: More Accurate NVFP4 Quantization with Adaptive Block Scaling”☆104Updated this week
- ☆255Updated last year
- High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.☆123Updated last year
- ☆270Updated this week
- ☆104Updated last year
- ☆84Updated 11 months ago
- Framework to reduce autotune overhead to zero for well known deployments.☆92Updated 3 months ago
- Distributed MoE in a Single Kernel [NeurIPS '25]☆174Updated this week
- An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).☆276Updated 5 months ago
- TritonBench: Benchmarking Large Language Model Capabilities for Generating Triton Operators☆100Updated 6 months ago
- ☆125Updated 4 months ago
- PyTorch bindings for CUTLASS grouped GEMM.☆180Updated 3 weeks ago
- Applied AI experiments and examples for PyTorch☆312Updated 4 months ago
- High-performance distributed data shuffling (all-to-all) library for MoE training and inference☆102Updated last week
- Collection of kernels written in Triton language☆174Updated 9 months ago
- Fast Hadamard transform in CUDA, with a PyTorch interface☆267Updated 2 months ago