machine learning engineer

Команда разрабатывает роботов доставки, которые ежедневно доставляют тысячи заказов в городской среде. Сервис переходит на архитектуру End-to-End на базе World Models для создания Embodied AI, способного планировать сложные манёвры в реальном мире.

• Проектировать и обучать массивные 3D/видеотокенизаторы и бэкбоны на базе Diffusion Transformers и Flow Matching.
• Строить пайплайны для распределённого обучения тяжёлых foundation-моделей на вычислительном кластере.
• Обучать чистое RL и политики IL + RL внутри замороженной латентной симуляции World Model.
• Интегрировать вспомогательные лоссы для задач 3D-детекции, сегментации и трекинга.
• Строить надёжный safety-контур поверх выходов модели и готовить конструкцию к реалтайм-инференсу на edge-девайсах робота.

• Экспертное владение JAX и PyTorch, глубокий практический опыт работы с современными фреймворками.
• Навыки масштабного распределённого обучения и опыт обучения тяжёлых моделей на мультинодовых кластерах.
• Глубокая математическая и ML-база, понимание непрерывных генеративных моделей и Deep RL.
• Умение писать оптимизированный код и доводить ресёрч до продакшна с жёсткими ограничениями реального времени.
• Будет плюсом опыт работы с Vision Foundation Models, генеративными видео- и image-моделями, синтезом лидарных облаков, Reinforcement Learning в LLM, продвинутой квантизацией тяжёлых трансформеров или диффузионных моделей, оптимизацией инференса с использованием C++, TensorRT, ONNX, CUDA, а также бэкграунд в Autonomous Driving, Motion Planning или Robotics.

• Расширенная медицинская страховка с первого месяца, включая стоматологию, чекапы и помощь за рубежом.
• Страхование родственников по системе 80/20.
• Доступ к внутренней образовательной платформе, менторству и оплата участия в профильных конференциях.
• Наличие спортзалов в крупных офисах и корпоративные спортивные программы.
• Гибкий график работы.
• Льготные жилищные займы для сотрудников российских офисов.
• Программы поддержки для родителей и детей.