Anyone Ai · Other · Live in: Argentina
Posted 2026-09-15 · Verified 2026-09-25
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 performance optimization and benchmarking Memory management across SBUF, PSUM, and HBM Tile-based computation... Read the full ad