Estados Unidos
Be Structured Technology Group
Waymo
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
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