Join NVIDIA as a Senior Machine Learning Engineer - Vector Search Focus

Are you a motivated and experienced Senior Machine Learning Engineer with a focus on Vector Search? NVIDIA’s RAPIDS Machine Learning team is looking for you! RAPIDS is an open-source library suite that combines the performance of modern GPUs with the convenience of Python APIs. We are continually enhancing the machine learning (ML) capabilities of RAPIDS and integrating them into critical infrastructure projects like model deployment and inference.

About the Role

As a Senior Machine Learning Engineer, you will join a team of skilled ML, HPC, and Software Engineers to develop revolutionary GPU-accelerated frameworks. You will contribute to and lead the feature development of RAPIDS ML Projects and other open-source initiatives, advancing the GPU Data Science and ML ecosystems.

Key Responsibilities

• Develop and refine innovative Vector Search and Machine Learning algorithms and pipelines.
• Integrate RAPIDS ML training and inference components into both enterprise and open-source software packages.
• Implement and enhance solutions based on systems with components in Python, Cython, C++, and CUDA.
• Contribute to open-source projects such as RAPIDS RAFT and cuML.
• Benchmark, profile, and optimize complex algorithms on different system architectures, from single nodes to high-scale distributed systems.
• Collaborate closely with Data Scientists, Solution Architects, and other groups to integrate, troubleshoot, and enhance our tools.

Qualifications

To be successful in this role, you should have:

• 8+ years of experience as an ML Engineer or Software Engineer, specializing in Vector Search, Nearest Neighbors, distance-based algorithms, or distributed systems.
• BS/MS in Computer Science, Computer Engineering, or a related engineering field (or equivalent experience).
• Strong modern C++ programming skills.
• Familiarity with Python.
• Knowledge of one distributed programming framework like Dask or Spark.
• Experience with build systems based around CMake and Docker.
• A deep commitment to creating robust, readable, well-documented, and high-performance code.
• An eagerness to explore new problem areas and apply your creativity to challenging and rewarding problems.

Preferred Skills

To stand out, you should have:

• Significant experience with Vector Databases such as Milvus, Pinecone, or frameworks like FAISS.
• Proficiency in CUDA.
• An understanding of build infrastructure and CI/CD technologies like CMake, Docker, Bash scripting, Jenkins, compilers, and linkers.
• Notable contributions and interactions with data science and ML open-source projects.
• Expert knowledge of Dask, Spark, or other distributed systems.

About NVIDIA

NVIDIA is widely regarded as one of the tech world’s most desirable employers. Our workforce comprises some of the most talented and hardworking individuals. If you are creative and passionate about developing cloud services, we want to hear from you!

The base salary range for this position is 220,000 USD - 419,750 USD. Your base salary will be determined by your location, experience, and the pay of employees in similar positions. Additionally, you will be eligible for equity and benefits.

NVIDIA accepts applications on an ongoing basis. We are committed to fostering a diverse work environment and are proud to be an equal opportunity employer. We highly value diversity in our current and future employees and do not discriminate based on race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status, or any other characteristic protected by law.

Application Information

Company: NVIDIA
Job Title: Senior Machine Learning Engineer

Apply now to become a part of NVIDIA’s innovative and dynamic team!