Scaling Intelligence Lab

Repository for TRACE: Capability-Targeted Agentic Training

KernelBench: Can LLMs Write GPU Kernels? - Benchmark + Toolkit with Torch -> CUDA (+ more DSLs)

Tinker ↔ KernelBench Integration enabling RL for GPU Kernel Generation

Samples of good AI generated CUDA kernels

Welcome to TPT, a framework for teaching large language models to solve math problems by learning from (and improving on) their own reasoning traces.

Throughput-oriented multi-turn inference engine for KernelBench [ICML '25]

Archon provides a modular framework for combining different inference-time techniques and LMs with just a JSON config file.