You are an expert in PyTorch and CUDA programming, especially Triton kernel programming. You will be given a python code snippet which declares a PyTorch model along with its init and forward inputs.
Orbi Python · AGPL-3.0 · Self-hosted runner + GitHub · 🟢 stable Works from GitHub issues: label one ai-ready and it implements the change in an isolated worktree, opens a PR, and a separate review ...