Ingeniero de Compiladores ML, Interfaz de Kernel Neuron

Los aceleradores AWS Machine Learning están a la vanguardia de la innovación de AWS. Trainium ofrece el mejor rendimiento de entrenamiento de ML de su clase con la mayor cantidad de teraflops (TFLOPS) de potencia de cómputo para ML en la nube. Todo esto es posible gracias al AWS Neuron Software Development Kit (SDK), que incluye un compilador de ML, el compilador Neuron Kernel Interface (NKI) y un tiempo de ejecución que se integra de forma nativa en marcos de ML populares como PyTorch y JAX.

Neuron Kernel Interface (NKI) es un lenguaje y compilador de bajo nivel para programar directamente las instancias AWS Trainium. Puede usar NKI para desarrollar, optimizar y ejecutar nuevos operadores directamente en el hardware, aprovechando al máximo los recursos de cómputo y memoria disponibles.

Explore NKI: - awsdocs-neuron.readthedocs-hosted.com/en/latest/nki/index.html

AWS Neuron es utilizado a gran escala por clientes como Epic Games, Snap, Airbnb, Autodesk, Amazon Alexa y Amazon Rekognition, entre muchos otros en una variedad de segmentos.

Más información sobre nuestra historia: - amazon.science/how-silicon-innovation-became-the-secret-sauce-behind-awss-success

Explore el producto: - aws.amazon.com/machine-learning/neuron/

Como Ingeniero de Compiladores ML en el equipo de Annapurna Neuron, desarrollará y escalará compiladores para manejar las cargas de trabajo de ML más grandes del mundo en AWS Trainium. Diseñará y enviará funciones en toda la pila del compilador, trabajará directamente con los equipos asociados de ML de Amazon para desbloquear sus cargas de trabajo y hará crecer sus habilidades junto con ingenieros experimentados. Desea trabajar cerca del hardware, ayudar a llevar nuevos productos y funciones al mercado y enviar código de alta calidad.

Comienza el día revisando el código de un compañero para una nueva función del compilador, dejando comentarios antes de que se fusione. A media mañana, trabaja con un equipo de modelos cuyo kernel no está alcanzando el rendimiento máximo en Trainium, y rastrea el cuello de botella a través de los pases del compilador. Después del almuerzo, escribe y prueba la función que diseñó la semana pasada, y luego la ejecuta en hardware para confirmar su corrección. El equipo se mueve rápidamente y usted está siempre en los detalles.

Sobre el equipo

Somos los equipos de compiladores detrás de AWS Neuron, el software que ejecuta modelos de machine learning en Amazon Trainium. Nuestro compilador de ML toma un modelo completo de PyTorch o JAX y compila el grafo para que se ejecute en todo el chip. Neuron Kernel Interface (NKI) es un lenguaje y compilador de bajo nivel que permite a los clientes programar Trainium directamente, hasta el cómputo y la memoria, para un nuevo operador o un kernel más rápido. Juntos, somos responsables del camino desde el modelo del framework hasta el silicio. Nuestra misión es hacer que Trainium sea rápido y totalmente programable.

Nuestros equipos son pequeños y ágiles, y hay oportunidades para aprender todo el tiempo. Trabajamos en estrecha colaboración con los equipos de hardware, tiempo de ejecución y frameworks, y con los equipos de ML de Amazon que utilizan nuestros compiladores.

Tener experiencia en Machine Learning, aceleradores de IA o rendimiento de Kernel es útil, pero no es un requisito.

Para ser considerado para este puesto, los candidatos deben residir cerca o estar dispuestos a reubicarse en Seattle o Cupertino.

Cualificaciones básicas:

Cualificaciones preferidas:

Amazon es un empleador que ofrece igualdad de oportunidades y no discrimina por motivos de condición de veterano protegido, discapacidad u otro estado legalmente protegido.

Candidatos del condado de Los Ángeles: Las responsabilidades del puesto incluyen: trabajar de forma segura y cooperativa con otros empleados, supervisores y personal; cumplir con los estándares de excelencia a pesar de las condiciones de estrés; comunicarse de manera efectiva y respetuosa con los empleados, supervisores y personal para garantizar un servicio al cliente excepcional; y cumplir con todas las leyes federales, estatales y locales y las políticas de la empresa. El historial penal puede tener una relación directa, adversa y negativa con algunas de las responsabilidades laborales materiales de este puesto. Estas incluyen las funciones y responsabilidades enumeradas anteriormente, así como la capacidad de cumplir con las políticas de la empresa, ejercer un juicio sólido, gestionar eficazmente el estrés y trabajar de forma segura y respetuosa con los demás, exhibir confiabilidad y profesionalismo, y salvaguardar las operaciones comerciales y la reputación de la empresa. De conformidad con la Ordenanza de Oportunidad Justa del Condado de Los Ángeles, consideraremos para el empleo a solicitantes calificados con antecedentes de arrestos y condenas.

Nuestra cultura inclusiva empodera a los amazónicos para ofrecer los mejores resultados a nuestros clientes. Si tiene una discapacidad y necesita una adaptación o ajuste en el lugar de trabajo durante el proceso de solicitud y contratación, incluido el apoyo para el proceso de entrevista o incorporación, visite amazon.jobs/content/en/how-we-hire/accommodations para obtener más información. Si el país/región en el que está solicitando no aparece en la lista, póngase en contacto con su socio de contratación.

EE. UU., CA, Cupertino - 165,200.00 - 223,600.00 USD anuales

EE. UU., WA, Seattle - 143,700.00 - 194,400.00 USD anuales

El rango salarial base para este puesto se indica a continuación. Su paquete de Amazon incluirá pagos de firma y unidades de acciones restringidas (RSU). La compensación final se determinará en función de factores como la experiencia, las cualificaciones y la ubicación. Amazon también ofrece beneficios integrales que incluyen seguro médico (médico, dental, oftalmológico, de prescripción, seguro básico de vida y AD&D y opción de planes de vida suplementarios, EAP, apoyo de salud mental, línea de asesoramiento médico, Cuentas de Gastos Flexibles, cobertura de reembolso de adopción y subrogación), igualación de 401(k), tiempo libre remunerado y licencia parental. Obtenga más información sobre nuestros beneficios en amazon.jobs/en/benefits.