Разработчик компиляторов для ML (Центр квантовых технологий)

Ищем старшего разработчика компиляторов для ML (C++/MLIR/OpenXLA/IREE) — человека, который умеет превращать графы ML моделей в быстрый код под реальные ускорители, не боится копаться в низкоуровневой оптимизации и уровнях IR. Будете работать на стыке hardware и software и участвовать в co-design всей архитектуры.

Обязанности

• разработка ML-компилятора на базе MLIR: custom MLIR-passes, IR, lowering, codegen
• оптимизации под фотонные ускорители: fusion, scheduling, cost model, memory planning
• профилирование производительности на реальных ML-моделях с учётом специфики фотонного ускорителя
• участие в проектировании и разработке софта системного уровня для фотонного ускорителя
• интеграция компиляторного тулчейна с PyTorch.

Требования

• отличные знания C++, уверенные знания в Python
• практический опыт с MLIR/OpenXLA/IREE/TVM/OpenVINO (не обязательно всё), умение разбираться в performance на низком уровне.
• понимание архитектур ускорителей (GPU/TPU/DSA).

Желательно:

• опыт оптимизации ML под GPU/TPU/другие ускорители
• практический опыт в проектировании ISA/опкодов/памяти/инструкций.

Условия

• формат гибридный, 3+ дня в офисе (м. Тульская)
• сильная команда (ex‑Samsung, ex‑Yandex)
• лаборатория и железо за соседним столом - изменения в прод катим быстро
• ежегодный пересмотр зарплаты и годовая премия
• корпоративный спортзал и зоны отдыха
• более 400 образовательных программ СберУниверситета для профессионального и карьерного развития
• расширенный ДМС, льготное страхование для семьи и корпоративная пенсионная программа
• гибкий дисконт по ипотечному кредиту, равный 1/3 ключевой ставки ЦБ
• бесплатная подписка СберПрайм+, скидки на продукты компаний-партнеров.