职业分类
软件开发工程师
描述
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…
原文语言：英语
更新