Nvidia Senior Machine Learning Engineer 

United States, Texas 

236569665

What you’ll be doing:

• Developing and improving innovative Vector Search and Machine Learning algorithms and pipelines.

• Integrating RAPIDS ML training and inference components into enterprise and open source software packages

• Implementing solutions based on systems that have components in Python, Cython, C++ and CUDA and improving those components.

• Contributing to open source projects, such as RAPIDS RAFT, cuML, and others. Benchmarking, profiling, and optimizing complex algorithms that have components in Python, Cython and C++ on different system architectures, from single node to high scale distributed systems.

• Working closely with Data Scientists, Solution Architects and other groups to integrate, triage, debug and develop feature roadmaps of our tools.

What we need to see:

• 8+ years of experience as a ML Engineer or Software Engineer, preferably building and maintaining either Vector Search, Nearest Neighbors or distance based algorithms and/or distributed systems.

• BS, MS in CS/CE or related engineering field (or equivalent experience)

• Strong Modern C++ programming skills

• Familiarity with Python.

• Knowledge of one distributed programming framework like Dask or Spark.

• Familiar with build systems based around CMake and Docker.

• You care deeply about robust, readable, well documented, well tested, high-performance code.

• Excited to learn, explore new problem areas, and apply your creativity to some of the most challenging and rewarding problems we have.

Ways to stand out from the crowd:

• Significant experience with Vector Databases like Milvus, Pinecone or frameworks like FAISS.

• Proficiency in CUDA.

• Understanding of build infrastructure and CI/CD related technologies such as CMake, Docker, Bash scripting, Jenkins, compilers, linkers.

• Significant contributions and interactions with data science and ML open source projects.

• Expert knowledge Dask, Spark or other distributed systems.