Technical Lead CUDA / Triton Kernel Development (AI)

TL;DR

Technical Lead CUDA / Triton Kernel Development (AI): Разработка и оптимизация низкоуровневых ядер для ускорения обучения и инференса LLM с акцентом на архитектуру GPU NVIDIA, работу с памятью и масштабирование на тысячи видеокарт. Фокус на снижении задержек (latency), увеличении пропускной способности (throughput) и глубокой интеграции с фреймворками PyTorch и TensorRT.

Локация: Работа возможна удаленно из любого города РФ или в гибридном формате.

Зарплата: 600 000 — 1 000 000 ₽/мес на руки.

Компания

Сбер — крупнейший банк в России и один из ведущих международных финансовых институтов, активно развивающий передовые AI-технологии и собственные большие языковые модели.

Что делать

• Лидировать направление разработки и оптимизации CUDA/Triton-ядер для LLM.
• Анализировать bottleneck’и в распределенных системах (multinode inference, sparse attention, mixture-of-experts).
• Тесно взаимодействовать с research-командами для реализации высокопроизводительных кастомных ядер.
• Участвовать в open-source инициативах, включая контрибьютинг в Triton, PyTorch и TensorRT.

Требования

• Локация: только РФ.
• 5+ лет опыта в low-level GPU программировании (CUDA C++, PTX/SASS, управление памятью и потоками).
• Глубокое понимание архитектуры NVIDIA GPU и опыт профилирования через Nsight.
• Опыт работы с PyTorch (custom ops, torch.compile) и TensorRT.
• 2+ года опыта управления технической командой (Lead/Tech Lead).

Хорошо, если есть

• Знание Triton и его отличий от чистой CUDA.
• Опыт работы с SGLang.
• Опыт оптимизации ядер типа FlashAttention или PagedAttention.

Культура и преимущества

• Ежегодный пересмотр зарплаты и годовая премия.
• Расширенный ДМС с льготным страхованием для семьи.
• Более 400 образовательных программ СберУниверситета.
• Корпоративная пенсионная программа и выгодная ипотека для сотрудников.
• Подписка СберПрайм+ и скидки у партнеров.