Senior Machine Learning Data Pipeline Engineer

NVIDIA · nvidia.com

US, CA, Santa Clara Full-time yesterday
Workplace
On-site
Location
US, CA, Santa Clara
Who can apply
Based in the United States: you usually need the right to work there

Check your CV against this job

Free · No signup · A 0–100 ATS match score and the keywords you're missing.

Check my CV free

Free to start. Already have an account? Log in

About the role

We are hiring a Senior Machine Learning Data Pipeline Engineer to invent and build the next generation of data pipelines for end-to-end autonomous driving foundation models. You will develop scalable pipelines and agentic workflows that help DNN teams proactively identify data gaps and efficiently discover, prepare, and deliver the right training data.

You will convert model limitations and data needs into reusable engineering solutions, integrating data infrastructure with LLMs, vision-language models, and retrieval techniques to automate data solutions.

What you'll be doing:

* Design, build, and own scalable data pipelines that automate data discovery, selection, and curation to meet DNN training data needs. * Develop agentic workflows that proactively identify model data needs, plan and implement mining tasks using data tools and model inference, and refine mining strategies through human review and downstream model feedback. * Deploy and operate these pipelines and workflows as reliable production systems, with workflow orchestration, monitoring, and failure recovery. * Partner with MLEs and data analysts to turn recurring data needs and model failure cases into reusable mining capabilities, define acceptance criteria, and improve scenario coverage through feedback.

What we need to see:

* BS or MS in Computer Science, Mathematics or Electrical Engineering, or a related quantitative field, or equivalent experience, plus 5+ years in software or data engineering. * Strong Python and SQL skills, shown through production services, APIs, or data pipelines. * Hands-on development of data solutions and optimization, including diagnosing DNN model performance and reliability issues. * Ability to develop and analyze automatic workflows that connect LLMs to tools and data services, including tool calling and workflow state management. * Solid understanding of deep learning, embeddings, and LLM/VLM inference, including how training datasets are prepared and consumed by DNN teams. * Shown experience in delivering data solutions starting from unclear requirements through to production, involving setting acceptance criteria alongside DNN engineers and data specialists and confirming outcomes.

Ways to Stand Out from the Crowd:

* Hands-on work on foundation model data and training pipelines, including smaller-scale models built from scratch. * A track record of shipping LLM/VLM-based mining, multimodal retrieval, or curation services. * Background in autonomous driving or robotics data, including temporal video, sensor metadata, scenario definitions, or model failure analysis.

#AutonomousVehicles

You will also be eligible for equity and benefits.

Applications for this job will be accepted at least until November 10, 2026.

This posting is for an existing vacancy.

NVIDIA uses AI tools in its recruiting processes.

NVIDIA is committed to fostering an inclusive work environment and proud to be an equal opportunity employer. As we highly value diversity in our current and future employees, we do not discriminate (including in our hiring and promotion practices) on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law.

About Machine Learning Engineer jobs in the United States

Open jobs
1,314 open Machine Learning Engineer jobs in the United States, 201 of them posted in the last 7 days. See all Machine Learning Engineer jobs in the United States
Typical advertised pay
$185,000 – $265,000 a year for the middle half of the 727 job ads from 305 employers that state a salary in US dollars (median $219,000). This is base pay as advertised, not total compensation. Machine Learning Engineer salary in the United States
Skills the job ads ask for most
  • Python 81%
  • LLMs 65%
  • AWS 35%
  • RAG 29%
  • PyTorch 29%
  • CI/CD 27%

Share of the 490 newest postings for the role that name each skill. Resume keywords for Machine Learning Engineer jobs

Tools and guides for your application

Get matched & apply with FindAJobAI

Upload your resume once. We score every job against your profile, tailor your resume and cover letter, and autofill the application.