Role OverviewEvaluate the quality, correctness, and hardware-appropriateness of Neuron Kernel Interface (NKI) development tasks for training and evaluating AI models. Assess CUDA→NKI migration fidelity and Trainium-specific performance-optimization quality.
What You Will Do
Work independently and asynchronously to improve AI model performance. Collaborate with AI research teams to ensure consistency and coverage across datasets.
Why It Might Be a Fit
Must have 2+ years of experience developing or optimizing kernels using NKI targeting AWS Trainium/Inferentia2 hardware. Strong understanding of NKI-specific development patterns and memory-hierarchy management.
Requirements
- 2+ years of experience developing or optimizing kernels using NKI targeting AWS Trainium/Inferentia2 hardware
- Strong understanding of NKI-specific development patterns and memory-hierarchy management
- Experience assessing CUDA→NKI migration quality
- Familiarity with Trainium-specific performance profiling
- Experience defining or evaluating cross-platform numerical-correctness standards
]]>