Forge CLI

AI-powered CUDA kernel optimization. Generate optimized GPU kernels that outperform torch.compile().

Install

``bash
npm install -g @rightnow/forge-cli
`

Or download binaries from GitHub Releases.

Usage

`bash


Launch optimization wizard

forge

Optimize a KernelBench task

forge optimize --task 1

Optimize a HuggingFace model

forge optimize --huggingface Qwen/Qwen2-0.5B

Optimize custom PyTorch code

forge optimize --file my_model.py
``

Features

- Swarm Agents - Multiple AI agents collaborate to find optimal kernels
- Evolutionary Search - MAP-Elites explores diverse optimization strategies
- CUDA & Triton - Generate kernels in both formats
- HuggingFace Support - Optimize transformer models directly

Requirements

- Node.js 18+
- Forge account (sign up)

Links

- Website
- Documentation
- GitHub

License

MIT