AI product

KForge

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.

Visit site Mentioned in 1 video ↓

What KForge is used for

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.

Videos mentioning KForge

1 in the library.