@rightnow/forge-cli
v0.1.3-a
Published
AI-powered CUDA kernel optimization CLI
Maintainers
Readme
Forge CLI
AI-powered CUDA kernel optimization. Generate optimized GPU kernels that outperform torch.compile().
Install
npm install -g @rightnow/forge-cliOr download binaries from GitHub Releases.
Usage
# 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.pyFeatures
- 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
License
MIT
