AMD-AGI/Apex

67 stars · Last commit 2026-05-29

Agents, and RL environment, for optimizing GPU kernels on AMD ROCm using LLM agents. Benchmarks LLM serving workloads end-to-end, profiles bottleneck kernels, optimizes them via Claude Code or Codex, and scores on compilation, correctness, and speedup.

README preview

# Apex — GPU Kernel Optimization Pipeline

> **License Notice**
>
> Copyright © 2025 Advanced Micro Devices, Inc. All rights reserved.
>
> This project is licensed under the **MIT License**. You are free to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the software, subject to the following conditions:
>
> - The above copyright notice and this license notice must be included in all copies or substantial portions of the software.
> - The software is provided **"as is"**, without warranty of any kind, express or implied, including but not limited to the warranties of merchantability, fitness for a particular purpose, and non-infringement.
>
> **Third-party dependencies** (ROCm libraries, vLLM, Triton, Magpie, etc.) are governed by their own respective licenses. End users are responsible for reviewing and complying with the licensing terms of any dependencies used in conjunction with this software.
>
> **AI Agent notice** — This software orchestrates third-party AI coding agents to perform kernel optimization:
>
> | Agent | Provider | What you need |
> |---|---|---|
> | **Claude Code** | Anthropic | An active [Anthropic account](https://console.anthropic.com/) and acceptance of [Anthropic's usage policies](https://www.anthropic.com/legal/usage-policy) |
> | **OpenAI Codex** | OpenAI | An active [OpenAI account](https://platform.openai.com/) and acceptance of [OpenAI's terms of service](https://openai.com/policies/terms-of-use) |
> | **Cursor Agent** | Cursor | An active [Cursor](https://cursor.com/) subscription with agent mode enabled |

View full repository on GitHub →