Custom AWS Transform agent that migrates PyTorch/Triton kernels to AWS Trainium NKI (@nki.jit) and compiles, numerically verifies, and profiles every candidate on a real Trainium device before opening a PR.
mcp pytorch triton neuron nki code-migration aws-transform llm-agent kernel-optimization aws-trainium amazon-bedrock-agentcore gpu-to-trainium
-
Updated
Jul 16, 2026 - Python