Senior Machine Learning Engineer

Вы мотивированный и опытный старший инженер по машинному обучению, специализирующийся на поиске по векторам? Команда машинного обучения RAPIDS в NVIDIA ищет вас! RAPIDS - это набор библиотек с открытым исходным кодом, который сочетает производительность современных GPU с удобством Python API. Мы постоянно улучшаем возможности машинного обучения (ML) RAPIDS и интегрируем их в критически важные инфраструктурные проекты, такие как развертывание моделей и инференс.

В качестве старшего инженера по машинному обучению вы присоединитесь к команде квалифицированных инженеров по ML, ВПК и программному обеспечению для разработки революционных фреймворков с ускорением на GPU. Вы будете вносить вклад и руководить разработкой функций проектов RAPIDS ML и других инициатив с открытым исходным кодом, развивая экосистемы Data Science и ML на базе GPU.

Ключевые обязанности:

• Разработка и совершенствование инновационных алгоритмов и конвейеров поиска по векторам и машинного обучения.
• Интеграция компонентов обучения и инференса RAPIDS ML в корпоративные и программные пакеты с открытым исходным кодом.
• Реализация и улучшение решений на основе систем с компонентами на Python, Cython, C++ и CUDA.
• Вклад в проекты с открытым исходным кодом, такие как RAPIDS RAFT и cuML.
• Тестирование, профилирование и оптимизация сложных алгоритмов на разных системных архитектурах, от одиночных узлов до масштабируемых распределенных систем.
• Тесное сотрудничество с учеными данных, архитекторами решений и другими группами для интеграции, устранения неполадок и улучшения наших инструментов.

Для успешной работы в этой должности вам следует иметь:

• 8+ лет опыта работы в качестве инженера по ML или инженера-программиста, специализирующегося на поиске по векторам, ближайших соседях, алгоритмах на основе расстояний или распределенных системах.
• Степень BS/MS в области компьютерных наук, компьютерной инженерии или в смежной инженерной области (или эквивалентный опыт).
• Сильные навыки программирования на современном C++.
• Знание Python.
• Знание одной из распределенных программных сред, таких как Dask или Spark.
• Опыт работы с системами сборки, основанными на CMake и Docker.
• Глубокая приверженность созданию надежного, читаемого, хорошо документированного и высокопроизводительного кода.
• Желание исследовать новые области проблем и применять свои творческие способности к сложным и полезным задачам.

Чтобы выделиться, вам следует иметь:

• Значительный опыт работы с векторными базами данных, такими как Milvus, Pinecone или фреймворками, такими как FAISS.
• Опыт работы с CUDA.
• Понимание инфраструктуры сборки и технологий CI/CD, таких как CMake, Docker, скрипты Bash, Jenkins, компиляторы и компоновщики.
• Значительный вклад и взаимодействие с проектами с открытым исходным кодом по науке о данных и машинному обучению.
• Экспертные знания Dask, Spark или других распределенных систем.

NVIDIA широко признана одним из самых желанных работодателей в мире технологий. Наша рабочая сила состоит из одних из самых талантливых и трудолюбивых людей. Если вы творческий человек и страстно относитесь к разработке облачных сервисов, мы хотим вас услышать!

Базовый оклад для этой должности составляет от 220 000 до 419 750 долларов США. Ваш базовый оклад будет определяться вашим местонахождением, опытом и уровнем оплаты сотрудников на аналогичных должностях. Кроме того, вы будете иметь право на акции и льготы.

NVIDIA принимает заявки на постоянной основе. Мы стрем