Назад
12 часов назад

GPU/NPU Parallel Computing Systems Developer (Autonomous Transport)

Формат работы
onsite
Тип работы
fulltime
Грейд
middle
Страна
Russia

Мэтч & Сопровод

Покажет вашу совместимость и напишет письмо

Описание вакансии

Текст:
/

TL;DR

Разработчик систем параллельных вычислений на GPU/NPU (AI/Autonomous Transport): Оптимизация и распараллеливание алгоритмов машинного обучения для систем автономного управления с акцентом на аппаратные особенности платформ и эффективность вычислений. Фокус на исследовании архитектур параллельного исполнения, конвертации моделей в ONNX и их запуске на TensorRT/TFLite.

Локация: Офис (Москва или Санкт-Петербург)

Компания

Yandex разрабатывает технологию автономного управления транспортом, включая беспилотные автомобили и роботов-доставщиков.

Что делать

  • Оптимизировать алгоритмы с учетом аппаратных особенностей различных платформ.
  • Автоматизировать тестирование алгоритмов и их адаптацию к целевому оборудованию.
  • Исследовать современные платформы параллельных вычислений.
  • Подбирать и реализовывать оптимальную архитектуру параллельного исполнения алгоритмов.

Требования

  • Опыт работы с PyTorch или TensorFlow, конвертацией ML-моделей в ONNX.
  • Опыт запуска моделей на TensorRT, ONNX Runtime, TFLite или аналогичных фреймворках.
  • Владение CUDA/OpenCL для оптимизации и распараллеливания вычислений.
  • Уверенное знание C++, Python, классических структур данных и алгоритмов.
  • Знание линейной алгебры и теории вероятностей.

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

  • Опыт решения прикладных задач с помощью Machine Learning или Deep Learning.
  • Опыт написания кастомных плагинов для ONNX или TensorRT на C++/CUDA.
  • Знакомство с методами сжатия, прунинга и квантизации нейросетей.
  • Знание дискретной математики и теории оптимизации.
  • Опыт индустриальной разработки, не связанной с ML.

Будьте осторожны: если работодатель просит войти в их систему, используя iCloud/Google, прислать код/пароль, запустить код/ПО, не делайте этого - это мошенники. Обязательно жмите "Пожаловаться" или пишите в поддержку. Подробнее в гайде →