Senior Software Engineer, ML Compilers, Edge TPU

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

Senior-инженер в команду EdgeTPU Compiler в Google. Занимается оптимизациями ML-компилятора для кастомного железа EdgeTPU и связывает ML-модели с исполнением на устройстве.

Задачи

• Проектировать и реализовывать оптимизации ML-компилятора для EdgeTPU
• Расширять фреймворки JAX и PyTorch для высокопроизводительной компиляции
• Диагностировать и устранять сложные проблемы корректности и производительности при деплое ML-моделей на EdgeTPU
• Предлагать и реализовывать фичи компилятора для устранения узких мест производительности и ограничений железа
• Работать с владельцами ML-моделей над архитектурой моделей для эффективного исполнения на EdgeTPU
• Вести проект от начала до конца, координируя кросс-функциональные команды и сроки

Требования

• Bachelor’s degree или эквивалентный практический опыт
• 5 лет опыта разработки на одном или нескольких языках программирования (например, C++)
• 3 года опыта тестирования, поддержки или запуска программных продуктов и 1 год опыта проектирования и архитектуры ПО
• 3 года опыта работы с компиляторами (оптимизация, параллелизация и т.п.)
• Предпочтительно: магистратура или PhD в Computer Science или смежной области
• Предпочтительно: опыт разработки компиляторов для архитектур на базе ускорителей
• Предпочтительно: опыт низкоуровневой разработки или сильный интерес к ней
• Предпочтительно: опыт работы с железом CPU, GPU или TPU
• Предпочтительно: опыт оптимизации инференса ML-моделей на устройстве

Условия

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

Навыки

• C++
• компиляторы
• JAX
• PyTorch
• ML
• CPU
• GPU
• TPU
• низкоуровневая разработка