Senior CUDA Engineer (Kandinsky)
По договоренности
Разрабатываем высокопроизводительные CUDA-операторы для PyTorch, обеспечивающие обучение и инференс мультимодальных моделей с максимальной утилизацией GPU-ресурсов. Фокус — низкоуровневая оптимизация, кастомные ядра, memory management и эффективная работа с новыми архитектурами GPU.
Обязанности
- разработка и оптимизация кастомных CUDA-операторов и расширений для PyTorch (C++/CUDA).
- профилирование и устранение узких мест в вычислительных ядрах (Nsight Compute, nvprof).
- оптимизация использования памяти (shared memory, registers, coalesced access, persistent kernels).
- реализация алгоритмов параллельных вычислений с учётом архитектурных особенностей современных GPU (Ampere, Hopper и новее).
- интеграция CUDA-оптимизаций в распределённые пайплайны обучения и инференса.
- тесная работа с командами Research и Distributed Learning для поддержки кастомных моделей и операторов.
Требования
- экспертный уровень C++ и CUDA.
- опыт оптимизации производительности для NVIDIA GPU.
- знание внутреннего устройства PyTorch (ATen, dispatcher, TensorIterator).
- навыки профилирования на GPU и поиска и устранения узких мест в реализации нейросетевых операторов.
- опыт работы с Mixed Precision и кастомными кернелами.
Бонус: Опыт с Triton, CUTLASS, cuBLASLt, NCCL; участие в open-source проектах PyTorch.
Условия
- комфортный современный офис рядом с м. Кутузовская
- формат работы гибрид
- ежегодный пересмотр зарплаты, квартальная и годовая премия
- корпоративный спортзал и зоны отдыха
- более 400 образовательных программ СберУниверситета для профессионального и карьерного развития
- программа адаптации и помощь руководителя на старте
- расширенный ДМС, льготное страхование для семьи и корпоративная пенсионная программа
- ипотека выгоднее до 7% для каждого сотрудника
- бесплатная подписка СберПрайм+, скидки на продукты компаний-партнеров
- вознаграждение за рекомендацию друзей в команду Сбера
Опубликована 6 дней назад
Похожие вакансии
Aston — аккредитованная аутсорсинговая IT-компания, работающая с 2007 года. Наш штат насчитывает 3400+ специалистов. Мы разрабатываем программное обеспечение и IT-решения в сферах Finance, Healthcare, eCommerce, IoT, Media, Big Data, ML и многом друг...
- Полный день
- Опыт от 1 года
- Москва
5 дней назад
Нахожусь в поисках Application Support Engineer в компанию, которая является лидером рынка в сфере доверительного управления и коллективных инвестиций. Предстоит работать над ключевыми для брокерского бизнеса системами. Что хотим видеть: Финансовая п...
- Полный день
- Опыт от 1 года
- Москва
4 дня назад
Мы динамично развивающаяся компания, специализирующаяся на поставках серверного и сетевого оборудования ведущих мировых брендов, а также собственной марки Xingfu Kele. Обязанности: Анализ ТЗ клиентов и подбор оптимальных конфигураций серверов (на баз...
- Полный день
- Опыт от 1 года
- Москва
6 дней назад
Продуктовая IT компания, которая занимается разработкой и поддержкой высоконагруженных проектов для крупных компаний, основная часть которых представляет собой развлекательные онлайн-сервисы, в поисках iOS Automation Engineer. Предстоит: Формировать ...
- Полный день
- Опыт от 1 года
- Москва
16 дней назад
200 000 - 350 000 ₽
Выше средней на 120%
Наш клиент - аккредитованная IT компания, разработчики ПО Приглашает на работу - ML-Engineer Требования: Высшее техническое образование Знание С++; Qt; Pyton Знание OpenCV, TensorFlow/PyTorсh, YOLO, MMDetection, Detectron2 Опыт разработки и внедрения...
- Полный день
- Опыт от 1 года
- Москва
18 дней назад
Мы — GigaChat Alignment. Делаем модель полезной и надёжной: SFT/DPO, дистилляция в маленькие модели, LoRA-сервис, метрики и валидирующие пайплайны. Быстро проверяем гипотезы, ускоряем обучение и выкатываем улучшения в прод — сначала для внутренних кл...
- Полный день
- Опыт от 1 года
- Москва
8 дней назад