广告合作
MCP.so
登录

Forge - GPU Kernel Optimization MCP Server

@RightNow-AI

代码仓库

基本信息

分类

其他

传输方式

stdio

作者

RightNow-AI

提交者

Jaber Jaber

关于 Forge - GPU Kernel Optimization MCP Server

Turn slow PyTorch into fast CUDA/Triton kernels. 32 parallel swarm agents optimize your code on real datacenter GPUs (B200, H200, H100, A100) with up to 14x speedup over torch.compile.

#gpu#cuda#triton#pytorch#optimization#kernel#machine-learning#deep-learning

Forge - GPU Kernel Optimization MCP Server 配置

使用下面的配置,将此服务器添加到你的 MCP 客户端。

{
  "mcpServers": {
    "forge": {
      "command": "npx",
      "args": [
        "-y",
        "@rightnow/forge-mcp-server"
      ]
    }
  }
}

Forge - GPU Kernel Optimization MCP Server 工具

7

Authenticate with the RightNow Forge GPU kernel optimization service. Opens the user's browser to sign in via the RightNow dashboard. Required before using any other Forge tools. If valid tokens already exist, verifies them without opening the browser.

Check your current Forge credit balance. Each single-GPU optimization costs 1 credit. Credits are only charged for successful optimizations (speedup >= 1.1x). Credits are lifetime (not monthly) and can be purchased on the dashboard.

Submit PyTorch code to Forge for GPU kernel optimization. Forge generates optimized Triton or CUDA kernels, evaluates them on real H100/A100 GPUs, and returns the best result with speedup metrics over torch.compile(max-autotune). USE THIS TOOL WHEN YOU SEE: - PyTorch custom operations (torch.autograd.Function, custom forward/backward) - Manual CUDA kernels that could be faster - Performance-critical tensor operations (attention, convolution, normalization, softmax) - Code with comments like "slow", "bottleneck", "optimize", "performance" - torch.compile() targets or triton.jit kernels - Any nn.Module with significant compute in forward() - Matrix multiplication, reduction, or scan operations - Custom loss functions with reduction operations - Fused operation opportunities (e.g., LayerNorm + activation) The tool streams real-time progress and blocks until optimization completes (1-10 minutes). Cost: 1 credit per optimization (only charged if speedup >= 1.1x). Requires authentication via forge_auth first.

Generate an optimized GPU kernel from scratch based on a specification. Use this when you need to create a new high-performance kernel without existing PyTorch code. Forge will generate a PyTorch baseline first, then optimize it into Triton or CUDA. Requires authentication via forge_auth first. Cost: 1 credit per generation.

Check the current status of a running or completed Forge optimization job.

Cancel a running Forge optimization job. Credits may be refunded if less than 20% of iterations completed.

List past Forge optimization sessions with their results. Useful for checking what has already been optimized.

概览

What is Forge - GPU Kernel Optimization?

Forge is an MCP server that optimizes PyTorch code into fast CUDA/Triton kernels using 32 parallel swarm agents. It is designed for AI developers and agents who want to accelerate GPU operations without manual kernel engineering. The server provides tools to optimize existing kernels, generate new ones from descriptions, and benchmark results on datacenter GPUs (B200, H200, H100, A100, L40S, T4).

How to use Forge - GPU Kernel Optimization?

Add the server to an MCP client (e.g., Claude) with the command claude mcp add forge-mcp -- npx -y @rightnow/forge-mcp-server. Authenticate via browser-based OAuth using the forge_auth tool, then call forge_optimize (to optimize PyTorch code) or forge_generate (to generate kernels from a description). Additional tools include forge_credits to check balance, forge_status to monitor jobs, forge_cancel to cancel jobs, and forge_sessions to list past sessions.

Key features of Forge - GPU Kernel Optimization

  • Optimize existing PyTorch code into faster Triton/CUDA kernels
  • Generate production-ready kernels from a natural language description
  • 32 parallel swarm agents (Coder+Judge pairs) compete for the fastest kernel
  • Real GPU benchmarking on datacenter hardware (B200, H200, H100, A100, L40S, T4)
  • Up to 14x faster than torch.compile(mode='max-autotune')
  • One-click browser-based OAuth, no API keys needed

Use cases of Forge - GPU Kernel Optimization

  • Accelerate custom PyTorch operations in AI models
  • Replace slow Python loops with optimized GPU kernels without manual CUDA coding
  • Generate kernels for novel operations described in natural language
  • Benchmark and compare kernel performance across multiple GPU architectures

常见问题

1

What authentication method does Forge use?

One-click browser-based OAuth; no API keys are needed.

2

What GPUs are used for benchmarking?

Datacenter GPUs: B200, H200, H100, A100, L40S, and T4.

3

How much does Forge cost?

Pay-as-you-go at $15/credit, with 25% off for 10+ credits. A free trial is included – optimize one kernel with no credit card required.

4

How many parallel agents work on optimization?

32 parallel swarm agents, each a Coder+Judge pair that competes to find the fastest kernel.

5

What tools does Forge provide?

Seven tools: forge_auth, forge_optimize, forge_generate, forge_credits, forge_status, forge_cancel, and forge_sessions.

评论

其他 分类下的更多 MCP 服务器

Blender

ahujasid

Open-source MCP to use Blender with any LLM

23.6K

MaxKB

1Panel-dev

🔥 MaxKB is an open-source platform for building enterprise-grade agents. 强大易用的开源企业级智能体平台。

22.0K

Hindsight

vectorize-io

Hindsight: Agent Memory That Learns

50.4K

🚀 Model Context Protocol (MCP) Curriculum for Beginners

microsoft

This open-source curriculum introduces the fundamentals of Model Context Protocol (MCP) through real-world, cross-language examples in .NET, Java, TypeScript, JavaScript, Rust and Python. Designed for developers, it focuses on practical techniques for building modular, scalable,

16.7K

FastMCP v2 🚀

jlowin

🚀 The fast, Pythonic way to build MCP servers and clients.

26.0K
广告合作