Ingeniero de Software - IA/ML, Aplicaciones AWS Neuron

AWS Neuron es el software completo para los aceleradores de aprendizaje automático a escala de nube AWS Inferentia y Trainium y los servidores Trn1 e Inf1 que los utilizan. Este puesto es para un ingeniero de software senior en el equipo de Aplicaciones de Aprendizaje Automático (ML Apps) para AWS Neuron. Este rol es responsable del desarrollo, habilitación y ajuste de rendimiento de una amplia variedad de familias de modelos de ML, incluyendo modelos de lenguaje a gran escala como GPT2, GPT3 y más allá, así como difusión estable, Vision Transformers y muchos más.

El equipo de ML Apps trabaja codo a codo con arquitectos de chips, ingenieros de compiladores e ingenieros de tiempo de ejecución para crear, construir y ajustar soluciones de entrenamiento distribuido con Trn1. Es imprescindible tener experiencia en el entrenamiento de estos grandes modelos utilizando Python. FSDP, Deepspeed y otras bibliotecas de entrenamiento distribuido son fundamentales para esto y extender todo esto para el sistema basado en Neuron es clave.

Responsabilidades clave del puesto
Este rol ayudará a liderar los esfuerzos para incorporar soporte de entrenamiento e inferencia distribuidos en Pytorch, Tensorflow usando XLA y las pilas de compilador y tiempo de ejecución de Neuron. Este rol ayudará a ajustar estos modelos para asegurar el máximo rendimiento y maximizar la eficiencia de los mismos ejecutándose en el silicio de AWS Trainium e Inferentia del cliente y los servidores TRn1, Inf1. Un sólido desarrollo de software y conocimientos de ML son fundamentales para este rol.

Sobre el equipo
Acerca de nosotros
Cultura de equipo inclusiva
Aquí en AWS, abrazamos nuestras diferencias. Estamos comprometidos a seguir promoviendo nuestra cultura de inclusión. Tenemos diez grupos de afinidad liderados por empleados, que llegan a 40,000 empleados en más de 190 capítulos a nivel mundial. Tenemos ofertas de beneficios innovadoras, y organizamos experiencias de aprendizaje anuales y continuas, incluyendo nuestras Conversaciones sobre Raza y Etnicidad (CORE, por sus siglas en inglés) y las conferencias AmazeCon (diversidad de género). La cultura de inclusión de Amazon se refuerza dentro de nuestros 16 Principios de Liderazgo, que recuerdan a los miembros del equipo que busquen perspectivas diversas, aprendan y sean curiosos, y se ganen la confianza.

Equilibrio trabajo/vida personal
Nuestro equipo valora mucho el equilibrio entre el trabajo y la vida personal. No se trata de cuántas horas pasas en casa o en el trabajo; se trata del flujo que estableces que aporta energía a ambas partes de tu vida. Creemos que encontrar el equilibrio adecuado entre tu vida personal y profesional es fundamental para la felicidad y la satisfacción a largo plazo. Ofrecemos flexibilidad en las horas de trabajo y te animamos a encontrar tu propio equilibrio entre tu vida laboral y personal.

Mentoría y crecimiento profesional
Nuestro equipo está dedicado a apoyar a los nuevos miembros. Tenemos una amplia mezcla de niveles de experiencia y antigüedad, y estamos construyendo un entorno que celebra el intercambio de conocimientos y la mentoría. Nos importa tu crecimiento profesional y nos esforzamos por asignar proyectos basados en lo que ayudará a cada miembro del equipo a convertirse en un profesional más completo y a asumir tareas más complejas en el futuro.

Estamos abiertos a contratar candidatos para trabajar en una de las siguientes localidades:

Cupertino, CA, USA | Seattle, WA, USA

Requisitos básicos

- 3+ años de experiencia profesional en desarrollo de software (no en pasantías)
- 2+ años de diseño o arquitectura (patrones de diseño, fiabilidad y escalado) de experiencia con sistemas nuevos y existentes (no en pasantías)
- Experiencia en programación con al menos un lenguaje de programación de software

Requisitos preferidos

- 3+ años de experiencia completa en el ciclo de vida del desarrollo de software, incluyendo estándares de codificación, revisiones de código, gestión de control de versiones, procesos de construcción, pruebas y experiencia en operaciones
- Licenciatura en Ciencias de la Computación o equivalente

Amazon está comprometido con un lugar de trabajo diverso e inclusivo. Amazon es un empleador de igualdad de oportunidades y no discrimina por raza, origen nacional, género, identidad de género, orientación sexual, estado de veterano protegido, discapacidad, edad u otro estado legalmente protegido. Para las personas con discapacidades que quisieran solicitar una adaptación, por favor visite https://www.amazon.jobs/en/disability/us.

Nuestra compensación refleja el costo de la mano de obra en varios mercados geográficos de los EE. UU. El pago base para este puesto oscila entre $115,000/año en nuestro mercado geográfico más bajo hasta $223,600/año en nuestro mercado geográfico más alto. El pago se basa en una serie de factores, incluyendo la ubicación del mercado y puede variar dependiendo del conocimiento, habilidades y experiencia relacionados con el trabajo. Amazon es una empresa de compensación total. Dependiendo del puesto ofrecido, el capital, los pagos de inicio de sesión y otras formas de compensación pueden ser proporcionados como parte de un paquete de compensación total, además de una gama completa de beneficios médicos, financieros y/o otros beneficios. Para obtener más información, por favor visite https://www.aboutamazon.com/workplace/employee-benefits. Los solicitantes deben aplicar a través de nuestro sitio de carreras internas o externas.