Logo of Huzzle

Senior Machine Learning Engineer

image

NVIDIA

5d ago

  • Job
    Full-time
    Senior (5-8 years) - Expert / Leadership (9+ years)
  • Software Engineering
  • $220K - $419.8K
  • Santa Clara
  • Quick Apply

AI generated summary

  • You: 8+ yrs ML/software engineering exp, C++, Python, distributed systems, build systems, high-performance code, eager to learn and problem solve. Stand out with Vector Databases, CUDA, CI/CD, open source contributions.
  • You will develop and enhance Vector Search and ML algorithms, integrate RAPIDS components, contribute to open source projects, optimize algorithms, and collaborate with cross-functional teams.

Requirements

  • 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.

Responsibilities

  • 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.

FAQs

What is the main focus of this Senior Machine Learning Engineer position?

The main focus of this role is on Vector Search within the RAPIDS Machine Learning team at NVIDIA.

Manufacturing & Electronics
Industry
10,001+
Employees
1993
Founded Year

Mission & Purpose

Since its founding in 1993, NVIDIA (NASDAQ: NVDA) has been a pioneer in accelerated computing. The company’s invention of the GPU in 1999 sparked the growth of the PC gaming market, redefined computer graphics, ignited the era of modern AI and is fueling the creation of the metaverse. NVIDIA is now a full-stack computing company with data-center-scale offerings that are reshaping industry.