gpu-to-trainium

Here is 1 public repository matching this topic...

aws-samples / sample-ATX-custom-NKI-Agent

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 Sep 1, 2026
Python

Add this topic to your repo

To associate your repository with the gpu-to-trainium topic, visit your repo's landing page and select "manage topics."

Learn more