GPU/NPU Parallel Computing Systems Developer (Autonomous Transport)
Мэтч & Сопровод
Покажет вашу совместимость и напишет письмо
Описание вакансии
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, прислать код/пароль, запустить код/ПО, не делайте этого - это мошенники. Обязательно жмите "Пожаловаться" или пишите в поддержку. Подробнее в гайде →