occupation
Software Developers
description
Convert PyTorch and Triton kernels to optimized NKI kernels for AWS Trainium and Inferentia. Use when the user says "convert this kernel to NKI", "optimize for Trainium", "PyTorch to NKI", "Triton to NKI", "rewrite this kernel for Inferentia", or "make this…
updated