Anyone AI is seeking experienced AWS Trainium / Neuron Kernel Interface engineers for a remote, part-time consulting project focused on evaluating and improving kernel development tasks for AI workloads.
You will review NKI kernels on Trainium or Inferentia2, migrate CUDA to NKI, optimize performance, and assess memory and DMA strategies. Strong feedback and deep technical know-how are essential.