Senior Vector Search ML Systems Engineer (CUDA/C++)

NVIDIA Corporation; Santa Clara (CA); On-site; USD 184,000 - 288,000; Full time

Job summary

NVIDIA in Santa Clara, California, seeks an outstanding software engineer to advance unstructured data processing libraries cuVS and cuML, crafting optimized C++/CUDA and exposing to Python and Java. You will research, develop, benchmark, and implement high-performance vector preprocessing, indexing, and ML algorithms for large-scale systems.

You will collaborate with a team of engineers on performance analysis, cross-language integration, and contributions to open source projects, while

Qualifications

• BS, MS, or PhD in Computer Science, Data Science or AI, Applied Math, or related field (or equivalent experience).
• 8+ years of experience programming in C++ or ability to learn it, by moving from a similar language: C, Rust, Java
• Experience with ML: ML concepts (approximation vs exact, iterative vs. closed-form), writing ML algorithms, and using ML to solve problems.
• Strong analytical problem-solving skills, algorithms and mathematics fundamentals.
• Excellent software development skills: programming, debugging, performance analysis, and test design
• Ability to work independently and manage your own development efforts.
• Good communication and documentation habits.
• You care deeply about robust, readable, high-performance code
• Familiar with at least one parallel programming or concurrency framework, such as CUDA, OpenMP, OpenACC, Java concurrency, pthreads, etc

Responsibilities

• Analyze, design, and implement optimized GPU algorithms for large-scale vector search, databases, and machine learning.
• Drive performance analysis, benchmarking, and optimization of associated libraries.
• Collaborate with a multi-functional team to understand requirements and implement or improve solutions
• Developing and improving machine learning algorithms.
• Implementing solutions in C++, CUDA and Python.
• Contributing to open source projects (like cuML, cuVS, and RAFT)

Skills

C++

CUDA

Python

Java

ML concepts

Performance optimization

Education

Bachelor's degree in CS/related

MS/PhD preferred

Job description

NVIDIA in Santa Clara, California, seeks an outstanding software engineer to advance unstructured data processing libraries cuVS and cuML, crafting optimized C++/CUDA and exposing to Python and Java. You will research, develop, benchmark, and implement high-performance vector preprocessing, indexing, and ML algorithms for large-scale systems.

You will collaborate with a team of engineers on performance analysis, cross-language integration, and contributions to open source projects, while