Senior Deep Learning Compiler Engineer - PyTorch

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

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

Задачи

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

Требования

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

Условия

• Высококонкурентная зарплата
• Расширенный пакет льгот
• Гибкая рабочая среда, поддерживающая разнообразие и инклюзивность
• Базовая зарплата определяется с учётом локации, опыта и зарплат сотрудников на аналогичных позициях
• Для Польши: диапазон базовой зарплаты 292 500 PLN - 507 000 PLN

Навыки

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