Разработчик компиляторов для 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 ключевой ставки ЦБ * бесплатная подписка СберПрайм+, скидки на продукты компаний-партнеров.