Anyone AI Inc. is seeking experienced engineers to work remotely on evaluating and improving kernel development tasks for AI workloads using AWS Trainium and Neuron Kernel Interface (NKI).
You will review NKI kernels, migrate CUDA to NKI, optimize performance, and assess memory management and DMA scheduling to ensure idiomatic Trainium usage. Strong feedback and cross-backend numerical understanding are essential.
📌 Remote Part-Time NKI Kernel Engineer for AWS Trainium (Argentina)
🏢 Anyone AI
📍 Argentina
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.