Software Engineer - ML Infrastructure

Автоматический пересказ вакансии на русском. Полный текст на языке оригинала и форма отклика — на сайте работодателя.

Инженер ML-инфраструктуры в healthcare-компанию, разрабатывающую AI-инструменты визуализации. Отвечает за распределённое обучение, RL-инфраструктуру и продакшн-сервинг моделей.

Задачи

• Строить инфраструктуру распределённого обучения с параллелизмом и чекпоинтингом.
• Разрабатывать инфраструктуру reinforcement learning: генерация rollout, reward-модели, сбор опыта.
• Совместно с исследователями превращать экспериментальные workflow в продакшн-системы.
• Строить пайплайны загрузки и предобработки данных для мультимодальных датасетов.
• Улучшать сервинг моделей, canary-деплой, мониторинг и инструменты rollout.

Требования

• 6+ лет опыта в ML-инфраструктуре или распределённых системах.
• Уверенный Python и практический опыт продакшн-сервинга ML.
• Опыт Kubernetes и Docker с GPU-планированием, автоскейлингом и воспроизводимыми окружениями.
• Экспертиза в распределённом обучении на PyTorch или JAX, включая FSDP, DeepSpeed или аналоги.
• Опыт с RL или online-learning циклами, логированием, чекпоинтингом и оценкой.
• Широкий охват ML-инфраструктурного стека и опыт технического лидерства или менторства.

Условия

• Базовая зарплата: от 250 000 до 300 000 USD в год.
• Конкурентные опционы (equity).
• Офис в Сан-Франциско, Калифорния, США, 5 дней в неделю.
• Поддержка трансфера визы и спонсорство новых виз.

Навыки

• Python
• Kubernetes
• Docker
• PyTorch
• JAX
• FSDP
• DeepSpeed
• Reinforcement Learning
• Distributed Training
• ML Serving
• GPU Scheduling
• A/B Testing