Anyone AI is seeking experienced engineers to evaluate and improve kernel development tasks for AI workloads on AWS Trainium and Inferentia2 hardware.
Desplácese hacia abajo para obtener una visión general completa de lo que requerirá este trabajo. ¿Es usted el candidato adecuado para esta posibilidad?
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. xghoner
This role focuses on kernel-level excellence.
📌 Nki Kernel Engineer – Aws Trainium Performance (Madrid)
🏢 Anyone AI
📍 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.