ML-разработчик в команду симуляции проезда автономного автомобиля в Автономный транспорт

Больше о команде

Ключевые критерии качества автономного автомобиля — безопасность, комфорт и эффективность. Оценивать их только на реальных дорогах долго, дорого, а для некоторых сценариев практически невозможно. Поэтому в Яндексе мы разработали систему симуляции, которая позволяет тестировать новые версии компонент автопилота в виртуальной среде и быстро оценивать, как меняется поведение автомобиля в разных дорожных ситуациях, погодных условиях и на разных территориях.

Достоверность таких тестов определяется тем, насколько реалистично окружающая среда реагирует на действия автономного автомобиля. Для этого мы используем ML-модель, которая предсказывает движение других участников дорожного движения: машин, мотоциклов и пешеходов. Над этой моделью и предстоит работать будущему участнику нашей команды.

Симулятор используют много команд, поэтому наши улучшения напрямую влияют на скорость и качество разработки автономного автомобиля Яндекса.

Какие задачи вас ждут:

Улучшение качества и разработка новых функций модели

Вам предстоит работать над повышением базового качества нейросети для генерации траекторий машин и пешеходов. Задачи включают улучшение датасета (сбор, фильтрацию и предобработку данных), архитектурные доработки моделей, оптимизацию процесса обучения и инференса, а также исследование возможностей внедрения inpainting, различных conditioning и guidance для решения продуктовых задач. Удачные наработки затем нужно будет внедрять в продакшен своими силами и силами бэкенд-команды.

Разработка фреймворка приемки моделей

Вторым фокусом будет участие в построении надежной и автоматизированной системы оценки качества и приемки новых версий модели. Нужно будет реализовывать в продакшен-системе и train-цикле моделей метрики из SOTA-статей, разрабатывать собственные метрики для оценки краевых случаев и внедрять краудсорсинг в процессы приемки для комплексной оценки моделей.

Больше об ML в Яндексе — в канале Yandex for ML

Мы ждем, что вы:

• Программировали на Python
• Работали с данными (сбор из сырых логов, формирование новых датасетов)
• Занимались генеративным CV
• Обучали нейросети на распределенных кластерах
• Участвовали в ML-проектах полного цикла с привязкой к бизнес-потребностям

Будет плюсом:

• Имеете практический опыт обучения диффузионных моделей
• Внедряли модели в продакшен через FastAPI + Docker, Triton Inference Server или любой другой фреймворк
• Участвовали в соревнованиях по ML или программированию (ACM ICPC, Codeforces, Topcoder, Kaggle и т. д.)
• Умеете программировать на C++
• Знаете правила дорожного движения, водили автомобиль или мотоцикл