Anyone AI is seeking experienced engineers to evaluate and improve kernel development tasks for AI workloads on AWS Trainium and Inferentia2 hardware. You will work on NKI kernels, migrations from CUDA, and performance benchmarking to optimize memory usage and DMA scheduling. You will review Trainium-specific patterns, optimize for SBUF/PSUM/HBM hierarchies, and ensure idiomatic, hardware-aware implementations rather than direct GPU translations. This role focuses on kernel-level excellence.
📌 NKI Kernel Engineer (Comunidad de Madrid)
🏢 Anyone AI
📍 Comunidad de Madrid
Postulate a este anuncio
Muestra tus habilidades a la empresa, rellenar el formulario y deja un toque personal en la carta, ayudará el reclutador en la elección del candidato.