Senior Software Engineer, AI/ML, ML Training

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

Инженер в команду JAX Models and Performance (ML Training, Core ML) в Google. Разработка и оптимизация open-source высокопроизводительных реализаций JAX для LLM и diffusion-моделей на GCP AI Hypercomputer (TPU, GPU, CPU) в сотрудничестве с DeepMind и Core ML.

Задачи

• Писать и тестировать код разработки продуктов или систем
• Взаимодействовать с коллегами и стейкхолдерами через дизайн- и код-ревью для соблюдения лучших практик
• Поддерживать документацию и обучающие материалы, адаптировать контент по обратной связи
• Триажить проблемы продукта или системы, отлаживать и устранять их, анализируя источники проблем и влияние на hardware, сеть и качество сервиса
• Проектировать и реализовывать решения в одной или нескольких специализированных ML-областях, используя ML-инфраструктуру

Требования

• Bachelor’s degree или эквивалентный практический опыт
• 5 лет опыта разработки ПО на одном или нескольких языках программирования
• 3 года опыта тестирования, поддержки или запуска программных продуктов и 1 год опыта в дизайне и архитектуре ПО
• 3 года опыта со сквозным Machine Learning (деплой моделей, оценка моделей, обработка данных, отладка)
• 3 года опыта с генеративным AI или анализом производительности
• Предпочтительно: Master's degree или PhD в Computer Science или смежной области
• Предпочтительно: 5 лет опыта со структурами данных и алгоритмами
• Предпочтительно: 2 года опыта с ML-акселераторами (TPU, GPU) и их тюнингом производительности
• Предпочтительно: 2 года опыта с оптимизацией производительности Mixture of Experts (MoE) или diffusion-моделей
• Предпочтительно: опыт сборки и релиза программных артефактов (pip wheels, docker images и т.п.)

Условия

• Медицинская, стоматологическая, офтальмологическая страховка, страхование жизни и инвалидности
• Пенсионный план 401(k) с match от компании
• 20 дней отпуска в год (начисление 6.15 часа за период выплат в первые пять лет)
• 40 часов/год больничных (69 часов/год для Сиэтла), включая 5 дискреционных больничных дней
• Декретный отпуск 28–30 недель
• 18 недель отпуска по уходу за ребенком
• 13 оплачиваемых праздничных дней в год
• Бонус 15% от цели + equity
• Возможность выбрать предпочтительное место работы: Sunnyvale, CA или Kirkland, WA

Навыки

• JAX
• Machine Learning
• Generative AI
• LLM
• Diffusion models
• Mixture of Experts (MoE)
• TPU
• GPU
• GCP AI Hypercomputer
• Python
• Docker
• pip
• Data structures
• Algorithms
• Performance tuning