AI product
KForge is an AI-assisted kernel-generation tool from Gimlet Labs that autonomously produces optimized low-level kernels from PyTorch for heterogeneous hardware. Its multi-agent system uses shared memory to explore kernel designs, enforce correctness checks, and identify fast implementations across CUDA, ROCm, and Metal, supporting both training and inference without requiring manual kernel authoring or changes to PyTorch. The system is described as using hardware and workload profiles to search optimization designs; in the cited video, Gimlet Labs reported a 33% end-to-end speedup for GPT-OSS 12B on AMD hardware in under a week, building on existing optimized kernels.
1 use taken from transcripts — each links to the moment in the video.
An agent built at Gimlet Labs that generates optimized kernels for a wide range of hardware by taking hardware and workload profiles and solving optimization as a brute-force problem. In under a week it sped up GPT-OSS 12B on AMD hardware by 33% end-to-end, building on already-optimized kernels.
1 in the library.