AWS Trainium / NKI Kernel Expert at Anyone AI

Anyone AI is recruiting experienced AWS Trainium / Neuron Kernel Interface (NKI) engineers for a specialized project focused on evaluating and improving kernel development tasks for AI workloads. We’re looking for engineers with hands-on experience building or optimizing NKI kernels on AWS Trainium or Inferentia2 hardware who understand how Trainium’s architecture differs from traditional GPU programming. ### WHAT YOU’LL WORK ON You’ll review and evaluate technical tasks involving: - NKI kernel correctness and Trainium-specific development patterns - CUDA → NKI kernel migrations - Trainium…