Staff Software Engineer ML Platform (6244)

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

Инженер в команду AI Factory компании Shield AI. Задача — строить backend-сервисы ML-платформы: трекинг экспериментов, реестр моделей и артефактов, чтобы ML-инженеры могли воспроизводить результаты и быстрее выводить модели в продакшн.

Задачи

• Строить сервисы трекинга экспериментов и реестра моделей, интегрируя инструменты вроде MLflow
• Связывать точные версии датасетов, кода и конфигураций с training runs, чекпоинтами и версиями моделей
• Разрабатывать API и адаптеры, соединяющие training jobs, сервисы данных и хранилища артефактов
• Обеспечивать восстановление прерванных и упавших workflow через надёжную обработку состояния, ретраи и диагностику
• Делать чекпоинты, артефакты моделей и результаты оценки удобными для поиска и получения
• Работать напрямую с ML-инженерами, упрощая их workflow и снижая ручной учёт
• Поставлять протестированные сервисы с observability и совместимыми апгрейдами в облачных и customer-managed окружениях

Требования

• Опыт построения production backend-сервисов для ML-платформ или data-intensive workflow
• Сильные навыки software engineering на Python или сравнимом backend-языке
• Опыт проектирования API, моделей метаданных и персистентного состояния для сервисов, используемых другими инженерами
• Опыт работы с хранилищами, частичными отказами, ретраями и консистентностью в распределённых системах
• Понимание training runs, чекпоинтов и версионирования моделей
• Умение работать напрямую с ML-пользователями и поставлять поддерживаемое ПО, решающее их задачи

Условия

• Full-time, регулярная занятость
• Оплата в указанном диапазоне + бонус + бенефиты + equity
• Офисы и площадки в США, Европе, на Ближнем Востоке и в Азиатско-Тихоокеанском регионе
• Работа на стыке современного AI, распределённых систем, Kubernetes и defense

Навыки

• Python
• MLflow
• Weights & Biases
• Kubernetes
• object storage
• distributed systems
• API design
• model registry
• experiment tracking