Детали вакансии

Инженер по производительности GPU в Яндексе. Специализируется на оптимизации использования и производительности GPU для ключевых сервисов Яндекса. Включает формирование гипотез, оптимизацию, профилирование и разработку инструментов.

Мы управляем одним из самых дефицитных и дорогих ресурсов компании — графическими процессорами (GPU). Их эффективное использование критически важно для работы ключевых сервисов Яндекса. Наша миссия — обеспечить максимальную отдачу и эффект от каждой GPU-карты. Какие задачи ждут вас: • Повышение эффективности использования GPU Формулировать гипотезы и исследовать пути повышения эффективности использования GPU, участвовать во внедрении наиболее выгодных решений. Формулировать рекомендации и лучшие практики по улучшению производительности. • Оптимизация и профилирование Выявлять узкие места производительности и устранять их с помощью профилировщиков, оптимизировать доступ к памяти, задержки и пропускную способность. • Разработка средств диагностики Создавать и улучшать инструменты для быстрой идентификации и устранения проблем инфраструктуры, влияющих на эффективность использования, стабильность и скорость GPU-вычислений. • Исследование и внедрение современных решений Изучать новейшие подходы к организации инфраструктуры для обучения и инференса, оценивать их эффективность и внедрять в проекты. • Анализ архитектуры, тестирование, интеграция Взаимодействовать с разработчиками, ML-инженерами и системными архитекторами. Участвовать в оценке аппаратных решений и предлагать улучшения для будущих поколений GPU.

Мы ожидаем, что вы: • Знаете Python и имеете опыт системного программирования • Работали с фреймворком PyTorch • Оптимизировали производительность GPU-приложений и повышали эффективность использования GPU • Работали с GPU (NVIDIA) и CUDA • Применяли подходы к распараллеливанию для распределённого инференса или обучения Будет плюсом, если вы: • Владеете C/C++ или аналогичными низкоуровневыми языками • Работали с библиотеками для обучения LLM с помощью RL

В Яндексе есть спортклубы, книжный клуб, киберспортивное сообщество. Это не все бонусы — полный список здесь: https://yandex.ru/jobs/pages/benefits?utm_campaign=ya_nanimaet