Senior Math Libraries Engineers - Python APIs

Чи є ви самовмотивованим експертом з розробки програмного забезпечення, захопленим дизайном і розробкою Python API для математичних бібліотек? NVIDIA шукає талановитих людей, таких як ви, щоб приєднатися до нашої динамічної команди, допомагаючи розширити можливості наших високопродуктивних GPU-акселераційних численних бібліотек для Python-базованих фреймворків.

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

Оскільки наші математичні бібліотеки продовжують зростати та розширюватися на нові домени, ми прагнемо зробити ці досягнення доступними для всіх розробників в екосистемі Python.

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

• Кандидат чи доктор наук з комп'ютерних наук, прикладної математики або суміжної галузі науки чи інженерії (або еквівалентний досвід)
• Відмінні навички програмування Python і C++, включаючи розробку функціональних і тестів продуктивності
• Досвід розробки Python API та Pythonic коду
• Знання однієї чи більше з Numpy, CuPy, SciPy, JAX, PyTorch, TensorFlow
• 6+ років досвіду розробки, відлагодження та оптимізації високопродуктивних застосунків на паралельних обчислювальних платформах
• Сильне розуміння фундаментальних методів численних розрахунків і обчислень у науці, інженерії або глибокому навчанні
• Доведений досвід керівництва та керування проектами розробки програмного забезпечення в умовах Agile
• Відмінні навички співпраці, комунікації та документації

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

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

Базова заробітна плата для цієї ролі становить