CUDA-разработчик (инфраструктура и ускорение вычислений) в Автономный транспорт

Локация: Москва, офис

Компания

Разработка вычислительного ядра и инфраструктуры для направления автономного транспорта.

Что делать

• Оптимизировать математические алгоритмы с учётом аппаратных особенностей вычислительных платформ.
• Автоматизировать тестирование алгоритмов и их адаптацию к целевым платформам.
• Ускорять инференс ML-моделей через ONNX с использованием квантизации и оптимизации вычислительных графов.
• Учитывать возможности бортовых ускорителей при оптимизации ML-инференса.
• Разрабатывать механизмы планирования вычислений и инфраструктуру для запуска нагрузок на GPU.
• Настраивать мониторинг распределения вычислительных ресурсов.

Требования

• От 3 лет опыта разработки.
• Профессиональное владение CUDA и C++.
• Глубокое понимание модели исполнения GPU и иерархии памяти.
• Умение измерять и объяснять причины потери производительности.
• Способность доводить оптимизацию от идеи до измеримого результата.

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

• Опыт работы с TensorRT, Triton, cuDNN или собственными kernel.
• Опыт оптимизации пайплайнов обработки облаков точек или изображений.
• Знание Python и ML-фреймворков, включая PyTorch.

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

• Работа над вычислительным ядром для автономного транспорта.
• Совместная работа с ML-командой над ускорением моделей.
• Офисный формат работы в Москве.