Ingeniero/a Staff Machine Learning, Infraestructura

Estados Unidos

Mountain View, California, United States · San Francisco, California, United States

Traducción automática. Consulta el original.

Sobre la oportunidad

Waymo es una empresa de tecnología de conducción autónoma con la misión de ser el conductor más fiable del mundo. Desde sus inicios como el proyecto de coche autónomo de Google en 2009, Waymo se ha centrado en desarrollar el Waymo Driver (el conductor más experimentado del mundo) para mejorar el acceso a la movilidad y salvar miles de vidas que actualmente se pierden en accidentes de tráfico. El Waymo Driver impulsa el servicio de transporte autónomo de Waymo y también puede aplicarse a una amplia gama de plataformas de vehículos y casos de uso de productos. El Waymo Driver ha realizado más de diez millones de viajes sin conductor, gracias a su experiencia en conducción autónoma durante más de 100 millones de millas en carreteras públicas y decenas de miles de millones en simulación en más de 15 estados de EE. UU.

Únete a nuestro equipo de ingeniería de infraestructura de ML que impulsa simulaciones multiagente ultrarrealistas de vanguardia mediante modelos fundacionales. En este puesto, trabajarás en la intersección de la infraestructura de ML, los modelos fundacionales y la ingeniería de simulación, con un enfoque específico en la escritura de lógica de negocio y simulación de alto rendimiento en JAX/TensorFlow ejecutándose directamente en TPU para potenciar entornos realistas para el aprendizaje por refuerzo (RL).

Lo que harás

Diseñar, construir y optimizar entornos de simulación realistas y lógica de negocio que se ejecuten en TPU utilizando JAX y TensorFlow. Implementar y optimizar estrategias de paralelismo de datos y modelos a gran escala para entrenar y ejecutar modelos fundacionales en hardware TPU.

Colaborar estrechamente con los equipos de modelado para integrar modelos fundacionales en los flujos de trabajo de simulación.

Impulsar arquitecturas técnicas y diseños de sistemas desde la ingeniería de datos hasta la ejecución de la simulación para cumplir con los objetivos de negocio y rendimiento.

Analizar sistemas, identificar cuellos de botella en el rendimiento a través de aceleradores de ML y optimizar la velocidad de ejecución de extremo a extremo.

Traducir los objetivos de producto y de negocio en requisitos técnicos concretos y entregables del sistema.

Requisitos mínimos

Más de 6 años de experiencia profesional en ingeniería de software, con al menos 4 años centrados en infraestructura de machine learning (escalado, entrenamiento, optimización y despliegue de sistemas de ML a gran escala).

Experiencia directa en programación de ML en hardware TPU y GPU utilizando marcos de trabajo como JAX, PyTorch o TensorFlow.

Experiencia práctica demostrada en el escalado de modelos grandes utilizando paralelismo de modelos, paralelismo de datos o técnicas de entrenamiento distribuido.

Sólida comprensión de los modelos de ML de vanguardia (por ejemplo, transformadores autorregresivos) y competencia práctica con herramientas de perfilado de aceleradores de ML para diagnosticar cuellos de botella.

Capacidad demostrada para liderar de forma independiente iniciativas técnicas ambiguas de principio a fin y construir bibliotecas, flujos de trabajo y herramientas de desarrollo robustas.

Sólidas habilidades de comunicación verbal y escrita para colaborar eficazmente en equipos distribuidos y multifuncionales.

Requisitos preferentes

Experiencia práctica en aprendizaje por refuerzo (RL), transferencia Sim2Real o robótica.

Experiencia con marcos de trabajo de ML distribuidos y aceleradores como GPU/TPU.

Familiaridad con sistemas de conducción autónoma, simulaciones multiagente o modelado realista del mundo.

El rango salarial base esperado para este puesto a tiempo completo en ubicaciones de EE. UU. se indica a continuación. El salario inicial real se basará en factores relacionados con el puesto, incluida la ubicación exacta de trabajo, la experiencia, la formación y educación relevantes, y el nivel de habilidades. Tu reclutador podrá compartir más información sobre el rango salarial específico para la ubicación del puesto o, si el puesto puede realizarse de forma remota, el rango salarial específico para tu ubicación preferida, durante el proceso de contratación.

Los empleados de Waymo también son elegibles para participar en el programa de bonificación anual discrecional de Waymo, el plan de incentivos de capital y el generoso programa de beneficios de la empresa, sujeto a los requisitos de elegibilidad.

Rango salarial

$251,000—$310,000 USD

Idiomas

Ingeniero/a Staff Machine Learning, Infraestructura

Sobre la oportunidad

Lo que harás

Requisitos mínimos

Requisitos preferentes

Rango salarial

Tu próximo empleoya está buscándote.

Más oportunidades con un solo perfil

Sigue explorando

Ingeniero de Sistemas Criptográficos

Ingeniero de Software Criptográfico

Ingeniero de Sistemas Senior