Principal Engineer, Efficient GenAI

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

Principal-инженер в команду AI Models and Applications в AMD. Задача — делать обучение и инференс генеративных моделей эффективными и масштабируемыми на платформах AMD.

Задачи

• Предлагать и применять инновационные техники для обучения и инференса: новые архитектуры трансформеров, стратегии параллелизма на больших кластерах, speculative decoding, оптимальное KV-кэширование
• Реализовывать эффективные архитектуры генеративных моделей и демонстрировать выигрыш на платформах AMD
• Работать с open-source фреймворками и сообществами (PyTorch, JAX, vLLM, SGLang), интегрировать оптимизированные под AMD модели и библиотеки, публиковать training recipes
• Совместно с командами софта и железа оптимизировать end-to-end производительность текущих и будущих решений AMD
• Развивать применение agentic workflows для оптимизации и деплоя генеративных приложений на платформах AMD
• Публиковать и продвигать результаты на внешних площадках, включая крупные конференции
• Сотрудничать с исследователями внутри AMD, в индустрии и академии

Требования

• Глубокая техническая экспертиза в обучении и инференсе генеративных моделей, опыт с PyTorch, JAX, vLLM, SGLang, MuJoCo
• Сильная экспертиза в алгоритмических инновациях для эффективных генеративных приложений в обучении и инференсе
• Публикации в одной или нескольких областях: эффективные архитектуры моделей, оптимизированное обучение, инновационные стратегии параллелизма, low-precision training
• Плюсом будут публикации на NeurIPS, CVPR, ECCV, ICCV, ICML или ICLR
• Опыт продуктазации генеративных моделей и обучения foundation-моделей в масштабе
• Отличные навыки письменной и устной коммуникации и презентаций, умение координировать работу внутри и снаружи
• Несколько лет опыта в AI, deep learning и смежной разработке ПО
• PhD или магистратура по computer science, электротехнике, математике или смежной области

Условия

• Работа в команде высококлассных специалистов
• Передовые генеративные AI-алгоритмы и софт
• Влияние на то, как модели обучаются в масштабе и как инференс-решения обслуживают миллионы клиентов
• Пакет бенефитов AMD

Навыки

• PyTorch
• JAX
• vLLM
• SGLang
• MuJoCo
• LLM
• Generative AI
• Deep Learning
• KV-caching
• speculative decoding
• parallelism strategies
• low-precision training
• agentic workflows