Software Engineer, ML Accelerators, DeepMind

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

Инженер в команду DeepMind для разработки вычислительной платформы следующего поколения под задачи AI-инференса с фокусом на ML-компиляторы, аппаратное ускорение и SW/HW codesign.

Задачи

• Вести full-stack разработку ПО с фокусом на ML-компилятор и инфраструктуру инференса
• Проектировать и разрабатывать ML-компиляторный стек для связки AI-нагрузок с низкоуровневыми аппаратными операторами
• Проектировать и разрабатывать программную инфраструктуру для высокопроизводительного инференса
• Заниматься SW/HW codesign, оптимизацией ядер, тюнингом производительности и отладкой

Требования

• Степень бакалавра в Computer Science, смежной технической области или эквивалентный практический опыт
• Опыт разработки на Python и C++
• Опыт работы с аппаратными ускорителями для ML, ML-компиляторными стеками (JAX, PyTorch, XLA) и разработкой ядер
• Предпочтительно: PhD в Computer Engineering, Computer Science или смежной области
• Предпочтительно: 2 года опыта full-stack разработки систем для аппаратных ускорителей
• Предпочтительно: 2 года опыта совместной разработки гипермасштабируемого ПО и инженерных best practices
• Предпочтительно: опыт постановки, планирования и выполнения проектов в условиях неопределённости

Условия

• Зарплата $174000–$252000 в год + целевой бонус 15% + equity + benefits
• Возможность выбрать предпочтительное место работы: Mountain View, CA или San Francisco, CA
• Работа в команде учёных и инженеров Google DeepMind

Навыки

• Python
• C++
• JAX
• PyTorch
• XLA
• ML hardware accelerators
• kernel development
• ML compiler
• inference infrastructure
• SW/HW codesign
• performance tuning