Especialista en kernels de Trainium (NKI)

Evalúa la calidad, corrección y adecuación al hardware de las tareas de desarrollo de Neuron Kernel Interface (NKI) utilizadas para entrenar y evaluar los modelos de un laboratorio de IA de vanguardia. Evaluarás la fidelidad de la migración de CUDA a NKI, la calidad de la optimización de rendimiento específica para Trainium y los estándares de corrección numérica multiplataforma, y proporcionarás comentarios escritos claros basados en rúbricas.

Requisitos Básicos • Más de 2 años de experiencia práctica desarrollando o Poniendo a punto kernels utilizando Neuron Kernel Interface (NKI) dirigidos al hardware AWS Trainium/Inferentia2 • Gran comprensión de los patrones de desarrollo específicos de NKI: computación basada en tiles, gestión de la jerarquía de memoria SBUF/PSUM/HBM, restricciones de dimensiones de partición y orquestación de DMA • Experiencia demostrada evaluando la calidad de la migración de CUDA a NKI • Familiaridad con la perfiles de rendimiento específicos de Trainium (utilización de la canalización de NeuronCore, rendimiento del motor de tensores, cuellos de botella en el ancho de banda de la memoria) • Experiencia definiendo o evaluando estándares de corrección numérica multiplataforma (orden de acumulación de GPU frente a Trainium, comportamiento de redondeo, semántica de precisión mixta)

Requisitos Preferidos • Experiencia directa con AWS Neuron SDK, aspectos internos del compilador Neuron o contribuciones a bibliotecas de kernels NKI • Desarrollo previo de kernels CUDA o Triton • Familiaridad con las especificaciones de hardware de Trainium (arquitectura NeuronCore-v2, topología de SRAM en el chip, tipos de datos compatibles: FP32/BF16/FP8/INT8) • Experiencia realizando pruebas de rendimiento de cargas de trabajo de entrenamiento de ML en instancias Trn1/Trn2