Senior Machine Learning Engineer

Job expired!

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

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

Основні обов'язки:

Розробка та вдосконалення інноваційних алгоритмів і конвеєрів для векторного пошуку та машинного навчання.
Інтеграція компонентів тренування та інференції 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, компілятори та лінкери.
Вагомі внески та взаємодії з відкритими проектами з науки про дані та ML.
Відомі знання Dask, Spark або інших розподілених систем.

NVIDIA широко визнана як один з