Senior Deep Learning Compiler Engineer - PyTorch

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

Инженер-компиляторщик в команду Thunder — source-to-source компилятора для PyTorch на GPU NVIDIA. Работа над ускорением глубокого обучения через современные компиляторные стеки.

Задачи

• Проектировать, реализовывать, оптимизировать и поддерживать ключевые компиляторные технологии для ускорения глубокого обучения
• Проводить анализ производительности нагрузок на тысячах GPU и находить возможности оптимизации
• Работать совместно с командами компиляторов, библиотек и систем (nvFuser, TVM, XLA, CUDA)
• Внедрять новейшие исследования в практические open-source решения

Требования

• Степень бакалавра, магистра или PhD в Computer Science или смежной области (или эквивалентный опыт)
• 8+ лет релевантного опыта работы
• Уверенное владение Python и опыт построения сложных, хорошо протестированных систем
• Практический опыт с фреймворками глубокого обучения, такими как PyTorch или JAX
• Понимание концепций компиляторов: AST, промежуточные представления (SSA), анализ программ, генерация кода
• Отличные коммуникативные навыки для работы в распределённой open-source среде

Условия

• Конкурентная зарплата
• Расширенный соцпакет
• Гибкий график и инклюзивная среда
• Работа над open-source проектом Thunder

Навыки

• Python
• PyTorch
• JAX
• TorchDynamo
• TorchInductor
• CUDA
• TVM
• MLIR
• IREE
• nvFuser
• XLA
• компиляторы
• AST
• SSA
• генерация кода