Senior Math Libraries Engineers - Python APIs

Являетесь ли вы самоорганизованным экспертом по программному обеспечению, увлечённым проектированием и разработкой Python API для математических библиотек? NVIDIA ищет талантливых людей, таких как вы, чтобы присоединиться к нашей динамичной команде, помогая развивать возможности наших высокопроизводительных математических библиотек, ускоренных на GPU, для фреймворков на основе Python.

За последнее десятилетие Python стал основным языком программирования для профессионалов в области ИИ, науки о данных и высокопроизводительных вычислений (HPC) благодаря мощным фреймворкам, таким как TensorFlow и PyTorch. NVIDIA сыграла ключевую роль в этом преобразовании, предоставив фреймворки для глубокого обучения с поддержкой GPU, которые обеспечивают эффективные высокоуровневые интерфейсы программирования. Наши оптимизированные реализации позволяют разработчикам сосредоточиться на приложениях, используя передовые технологии ускорения на GPU.

Поскольку наши математические библиотеки продолжают расти и расширяться в новые области, мы стремимся сделать эти достижения доступными для всех разработчиков в экосистеме Python.

• Создание и разработка Python API для математических библиотек (например, линейная алгебра)
• Прототипирование интеграций разработанных API в целевые фреймворки
• Настройка производительности, оптимизация и тестирование численного программного обеспечения на различных архитектурах CPU и GPU
• Предоставление технического руководства и наставничество инженерам библиотек
• Тесное сотрудничество с отделом управления продуктами и другими внутренними и внешними партнёрами для понимания требований к функциям и производительности
• Выявление возможностей для улучшения производительности и удобства использования программного обеспечения

• Степень PhD или MSc в компьютерных науках, прикладной математике или смежной области науки или инженерии (или эквивалентный опыт)
• Отличные навыки программирования и проектирования программного обеспечения на Python и C++, включая функциональное и производственное тестирование
• Опыт разработки Python API и Python-кода
• Знание одной или нескольких библиотек: Numpy, CuPy, SciPy, JAX, PyTorch, TensorFlow
• Более 6 лет опыта разработки, отладки и оптимизации высокопроизводительных приложений на платформах параллельных вычислений
• Глубокое понимание фундаментальных численных методов и вычислений в науке, инженерии или глубоком обучении
• Подтвержденный опыт руководства и управления проектами разработки программного обеспечения в гибкой среде
• Сильные навыки сотрудничества, коммуникации и документооборота

• Опыт параллельного программирования с использованием GPU (CUDA или OpenCL), многопоточности или MPI
• Хорошие знания архитектуры аппаратного обеспечения CPU и/или GPU

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

Базовый диапазон заработной платы для этой роли составляет $180 000 - $339 250, определяется в зависимости от вашего местоположения, опыта и оплаты сотрудников на аналогичных должностях. Вы также будете иметь право на получение акций и льгот.

NVIDIA принимает заявки на постоянной основе. Мы стремимся к созданию разнообразной рабочей среды и гордимся тем, что являемся работодателем, предоставляющим равные возможности. Разнообразие высоко ценится в наших нынешних и будущих сотрудниках, и мы не практикуем дискриминацию (включая наши практики найма и продвижения) на основе расы, религии, цвета кожи, национального происхождения, пола, гендерного выражения, сексуальной ориентации, возраста, семейного положения, статуса ветерана, инвалидности или любой другой характеристики, защищенной законом.