GPU/NPU Parallel Computing Systems Developer (Autonomous Transport)

Описание

Локация: Офис (Москва или Санкт-Петербург)КомпанияYandex разрабатывает технологию автономного управления транспортом, включая беспилотные автомобили и роботов-доставщиков.Что делатьОптимизировать алгоритмы с учетом аппаратных особенностей различных платформ.Автоматизировать тестирование алгоритмов и их адаптацию к целевому оборудованию.Исследовать современные платформы параллельных вычислений.Подбирать и реализовывать оптимальную архитектуру параллельного исполнения алгоритмов.ТребованияОпыт работы с PyTorch или TensorFlow, конвертацией ML-моделей в ONNX.Опыт запуска моделей на TensorRT, ONNX Runtime, TFLite или аналогичных фреймворках.Владение CUDA/OpenCL для оптимизации и распараллеливания вычислений.Уверенное знание C++, Python, классических структур данных и алгоритмов.Знание линейной алгебры и теории вероятностей.Хорошо, если естьОпыт решения прикладных задач с помощью Machine Learning или Deep Learning.Опыт написания кастомных плагинов для ONNX или TensorRT на C++/CUDA.Знакомство с методами сжатия, прунинга и квантизации нейросетей.Знание дискретной математики и теории оптимизации.Опыт индустриальной разработки, не связанной с ML.