Estados Unidos
The MITRE Corporation
Waymo
Ingeniero Staff de 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 de mayor confianza del mundo. Desde sus inicios como el Proyecto de Coche Autónomo de Google en 2009, Waymo se ha centrado en construir el Conductor Waymo —El Conductor Más Experimentado del Mundo™— para mejorar el acceso a la movilidad y, al mismo tiempo, salvar miles de vidas que ahora se pierden en accidentes de tráfico. El Conductor Waymo impulsa el servicio de ride-hail totalmente autónomo de Waymo y también puede aplicarse a una gama de plataformas de vehículos y casos de uso de productos. El Conductor Waymo ha realizado más de diez millones de viajes solo para pasajeros, gracias a su experiencia conduciendo de forma autónoma 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 para avanzar en simulaciones multiagente ultrarrealistas de vanguardia utilizando 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 simulación y empresarial de alto rendimiento en JAX/TensorFlow ejecutándose directamente en TPUs para potenciar entornos realistas para el Aprendizaje por Refuerzo (RL).
Qué Harás
- Diseñar, construir y optimizar entornos de simulación realistas y lógica empresarial ejecutándose en TPUs utilizando JAX y TensorFlow. Implementar y optimizar estrategias de paralelismo de modelos y datos 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 los objetivos empresariales y de rendimiento.
- Analizar el rendimiento de los sistemas, identificar cuellos de botella en los aceleradores de ML y optimizar la velocidad de ejecución de extremo a extremo.
- Traducir los objetivos de producto y negocio en requisitos técnicos concretos y entregables del sistema.
Cualificaciones Mínimas
- +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 frameworks 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 extremo a extremo y construir bibliotecas, flujos de trabajo y herramientas para desarrolladores robustas.
- Excelentes habilidades de comunicación verbal y escrita para colaborar eficazmente en equipos distribuidos y multifuncionales.
Cualificaciones Preferidas
- Experiencia práctica en Aprendizaje por Refuerzo (RL), transferencia Sim2Real o Robótica.
- Experiencia con frameworks de ML distribuidos y aceleradores como GPU/TPU.
- Familiaridad con sistemas de conducción autónoma, simulaciones multiagente o modelado de mundo realista.
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 habilidad. Su reclutador podrá compartir más información sobre el rango salarial específico para la ubicación del puesto o, si el puesto se puede realizar de forma remota, el rango salarial específico para su ubicación preferida, durante el proceso de contratación.
Los empleados de Waymo también son elegibles para participar en el programa discrecional de bonificación anual de Waymo, el plan de incentivos de capital y el generoso programa de beneficios de la Compañía, sujeto a los requisitos de elegibilidad.
Rango Salarial
$251,000—$310,000 USD
Más oportunidades con un solo perfil
¿Quieres optar a más puestos sin repetir todo el proceso? Crea tu perfil para que te encuentren las empresas que usan Nort.
Crear mi perfil