Zoox разрабатывает технологии автономной мобильности и специально созданный автопарк для мобильности как услуги. Компания объединяет робототехнику, искусственный интеллект, автомобилестроение, электромобили и передовые системы восприятия, чтобы создавать транспорт, который делает поездки безопаснее, экологичнее и приятнее. Zoox стремится изменить представление людей о персональной мобильности с помощью полностью автономных автомобилей, созданных с учётом потребностей пассажиров.
Руководитель инженерного направления по оптимизации производительности ML в Zoox
Руководите оптимизацией производительности машинного обучения для автономных систем вождения Zoox. Развивайте масштабируемое обучение, ускорение на GPU и инференс с низкой задержкой в инженерных командах робототехники.
Описание
- Определять стратегическое направление и дорожную карту оптимизации обучения и инференса машинного обучения
- Повышать масштабируемость, надежность и производительность рабочих нагрузок автономного вождения
- Руководить эффективной ML-платформой, охватывающей обучение, валидацию, развертывание, оптимизацию и мониторинг моделей
- Оптимизировать крупномасштабное обучение моделей и инференс от начала до конца
- Повышать эффективность распределенного обучения, улучшая использование GPU, работу с памятью и обмен данными
- Применять квантизацию, прореживание, дистилляцию и другие методы сжатия моделей
- Обеспечивать инференс на борту автомобиля с низкой задержкой в условиях жестких ограничений реального времени и вычислительных ресурсов
- Формировать и мотивировать разнопрофильную инженерную команду через найм и лидерство
- Сотрудничать с командами исследований ML, разработки ПО, работы с данными и аппаратного обеспечения
- Формировать требования и направлять архитектурные решения
- Наставлять инженеров, создавая возможности для развития и предоставляя своевременную обратную связь
Требования
- Не менее 8 лет соответствующего опыта, включая 3 года или более управления инженерными командами
- Глубокая экспертиза в оптимизации производительности ML, включая распределенное обучение, параллелизм по данным, тензорам и конвейерам, FSDP/ZeRO, смешанную точность, CUDA, Triton, torch.compile, XLA, TVM, квантизацию и профилирование на GPU и встроенных ускорителях
- Опыт создания доступной ML-инфраструктуры для крупномасштабного обучения и высокопроизводительного обслуживания с низкой задержкой
- Опыт использования PyTorch или JAX с GPU для распределенного обучения моделей
- Опыт ускоренного на GPU инференса с использованием TensorRT, Ray Serve или сопоставимых фреймворков
- Подтвержденный опыт межфункциональной работы с командами исследований, продукта, аппаратного обеспечения и платформ
- Умение расставлять приоритеты, определять техническое направление и обеспечивать измеримый прирост производительности в разных подразделениях
Преимущества
- Оплачиваемый отпуск, включая больничные, отпускные дни и отпуск в связи с утратой близкого человека
- Неоплачиваемый отпуск
- Права на прирост стоимости акций Zoox
- Ограниченные паи Amazon (RSU)
- Медицинское страхование
- Страхование на случай необходимости долгосрочного ухода
- Страхование долгосрочной и краткосрочной нетрудоспособности
- Страхование жизни
- Возможный бонус при найме