Software Engineer, ML Infrastructure

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

Инженер в команду Infrastructure Engineering института Arc. Владеет данными и оценкой ML-инфраструктуры для виртуальных клеточных моделей: дашборд производительности моделей и пайплайн подготовки обучающих данных.

Задачи

• Владеть бэкендом Model Performance Dashboard и развивать его
• Совместно с ML-командой определять, проектировать и строить инструменты оценки моделей
• Перепроектировать пайплайн превращения single-cell данных в обучающие наборы и запускать его с AI-агентами
• Владеть lab-in-the-loop инфраструктурой, связывающей результаты моделей с планированием экспериментов
• Строить или покупать фреймворки валидации (тесты, evals, QC-гейты, политики) для кода, написанного агентами
• Обеспечивать быстрые и надёжные тренировки и оценку ML-моделей в масштабе
• Поддерживать multi-node обучение, разбирать сбои распределённого обучения, повышать throughput и GPU utilization (MFU)

Требования

• Bachelor’s degree и 8 лет релевантного опыта, либо Master’s degree и 6 лет опыта
• Опыт владения production data или ML-платформой end-to-end (пайплайны, хранение, версионирование) и определения её технического направления
• Умение проектировать и строить системы с AI-инструментами и агентами, делая их результат проверяемым через evals, QC-гейты или тесты
• Способность быстро осваивать незнакомую техническую область до уровня принятия проектных решений
• Работа напрямую со стейкхолдерами без PM-посредника
• Опыт диагностики и устранения проблем производительности в Python или PyTorch с помощью профайлеров и измерений: GPU utilization, память, загрузка данных
• Работа в офисе в Palo Alto минимум 3 дня в неделю

Условия

• Базовая зарплата $191,000–$237,000 в год
• Ежегодный дискреционный бонус
• Участие в Compound Cash Grant Program (многолетние денежные гранты)
• Долгосрочное финансирование и ресурсы уровня индустрии
• Гибридный формат: минимум 3 дня в неделю в офисе Palo Alto

Навыки

• Python
• PyTorch
• GPU
• MFU
• DDP
• FSDP
• NCCL
• SLURM
• AnnData
• h5ad
• single-cell data
• ML infrastructure
• AI agents
• evals