Atrex Kernel Agent v0.1.0 — Initial Open-Source Release

Atrex Kernel Agent (AKA) v0.1.0

Initial open-source release. AKA is an end-to-end Agent project for GPU kernel implementation, analysis, profiling, and iterative optimization. It helps an Agent turn PyTorch logic or an existing kernel into a high-performance GPU kernel through a structured, profile-driven workflow.

✨ Highlights

• Isolated optimization workspace: Automatically creates a dedicated workspace under /tmp/kernel_opt_<name>/.
• Hardware-aware targets: Looks up target hardware specs from the local gpu-wiki knowledge base and runs Roofline analysis to set auditable performance targets.
• Profile-driven optimization loop: Builds a correct baseline kernel first, then iterates — profile with ncu (NVIDIA) or rocprofv3 (AMD), extract bottleneck evidence, query gpu-wiki / reference projects / web sources, write an evidence-based plan, apply one optimization category, validate correctness and performance, record memory, and commit — until the Stop Conditions are met.
• Fully traceable: Records plans, profile artifacts, structured memory, reports, and Git commits for every accepted iteration.
• Multi-platform: Supports NVIDIA (ncu) and AMD (rocprofv3, wrapped by tools/profile_kernel.sh).
• Multiple Agent runtimes: Works with both Codex and Claude Code.

📦 Installation

./install.sh