Senior Research Engineer, ML Data Pipelines

3 Months ago • 10 Years + • Research & Development • Artificial Intelligence • $224,000 PA - $425,500 PA

Job Summary

Job Description

NVIDIA's GEAR group seeks a Senior Research Engineer to design, implement, and optimize large-scale ML data pipelines for training multimodal foundation models for humanoid robots (Project GR00T). Responsibilities include collaborating with researchers on data preprocessing, developing data labeling tools, monitoring data quality, and optimizing PyTorch data loading modules on GPU clusters. The ideal candidate possesses extensive experience in large-scale ML data pipelines, proficiency in Python and JavaScript/TypeScript, and experience with GPU clusters and job scheduling tools.
Must have:
  • 10+ years experience with large-scale ML data pipelines
  • Proficiency in Python data processing libraries and PyTorch
  • Experience with large-scale GPU clusters and job scheduling
  • JavaScript/TypeScript and React proficiency for frontend development
Good to have:
  • Master's or PhD in CS/Robotics/Engineering
  • Cloud infrastructure (AWS, Azure, GCP) and data store experience
  • Experience in autonomous driving or robotics
  • Tech Lead experience, open-source contributions
Perks:
  • Equity
  • Benefits

Job Details

NVIDIA is searching for a senior or principal engineer who specializes in designing and optimizing large-scale machine learning data pipelines in the Generalist Embodied Agent Research (GEAR) group. Our team is leading Project GR00T, NVIDIA’s moonshot initiative at building foundation models and full-stack technology for humanoid robots.

You will work with an amazing and collaborative research team that consistently produces influential works on multimodal foundation models, large-scale robot learning, embodied AI, and physics simulation. Our past projects include Eureka, VIMA, Voyager, MineDojo, MimicPlay, Prismer, and more. Your contributions will have a significant impact on our research projects and product roadmaps.

What you will be doing:

  • Design, implement, and optimize scalable ML data pipelines for training multimodal foundation models;

  • Collaborate closely with researchers to preprocess, transform, and manage large datasets for robot model training and evaluation;

  • Develop tools for labeling and curating multiple streams of sensor data;

  • Continuously monitor robot data collection process and evaluate data quality;

  • Implement and optimize PyTorch data loading modules for video processing and robot learning on large GPU clusters.

What we need to see:

  • Bachelor's Degree in Computer Science, Robotics, Engineering, or a related field;

  • 10+ years of full-time industry experience working with large-scale machine learning data pipelines;

  • Proficiency in JavaScript/TypeScript and React for building frontend applications, such as dashboards, visualization, and analysis tools;

  • Proficiency in Python data processing libraries. Hands-on model training experience in PyTorch, JAX, or Tensorflow;

  • Strong experience with large-scale GPU clusters, HPC environments, and job scheduling/orchestration tools (e.g., SLURM, Kubernetes).

Ways to stand out from the crowd:

  • Master’s or PhD’s degree in Computer Science, Robotics, Engineering, or a related field;

  • Strong experience with cloud infrastructure management (AWS, Azure, GCP) and data stores (Postgres, MySQL, ElasticSearch, Redis);

  • Experience at autonomous driving or robotics companies training machine learning models on massive datasets;

  • Demonstrated Tech Lead experience, coordinating a team of engineers and driving projects from conception to deployment.

  • Contributions to popular open-source frameworks.

NVIDIA is widely considered to be one of the technology world's most desirable employers. We have some of the most forward-thinking and productive people in the world. Please join us and be part of the forefront of developing general-purpose robots and large-scale foundation models!

The base salary range is 224,000 USD - 425,500 USD. Your base salary will be determined based on your location, experience, and the pay of employees in similar positions.

You will also be eligible for equity and benefits. NVIDIA accepts applications on an ongoing basis.

NVIDIA is committed to fostering a diverse 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.

Similar Jobs

Ciklum - Recruitment Consultant

Ciklum

Pune, Maharashtra, India (On-Site)
7 Months ago
Easygo - Software Development Engineer (Frontend)

Easygo

Melbourne, Victoria, Australia (On-Site)
2 Months ago
Highspot - Principal Frontend Web Engineer

Highspot

Seattle, Washington, United States (Hybrid)
6 Months ago
Eleven Labs - AI Safety Engineer (Back-End)

Eleven Labs

United Kingdom (Remote)
3 Months ago
Ello - Design Engineer (Mobile)

Ello

Canada (On-Site)
5 Months ago
ByteDance - Research Scientist in ML Systems

ByteDance

Seattle, Washington, United States (On-Site)
5 Months ago
NVIDIA - Deep Learning Computing Performance Architect

NVIDIA

Shanghai, Shanghai, China (Hybrid)
3 Months ago
Microsoft - Research Intern - Bioinformatics

Microsoft

Redmond, Washington, United States (On-Site)
3 Months ago
Samsung Semiconductor - Intern, Machine Learning Research Scientist

Samsung Semiconductor

San Jose, California, United States (Hybrid)
3 Months ago
NVIDIA - Senior Datacenter GPU Power Architect

NVIDIA

Santa Clara, California, United States (On-Site)
3 Months ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Easygo - Senior DevOps Engineer

Easygo

Melbourne, Victoria, Australia (On-Site)
2 Months ago
Anthology  Inc  - Senior Software Engineer I

Anthology Inc

Bengaluru, Karnataka, India (On-Site)
6 Months ago
Sporty Group - Frontend Engineering Team Lead

Sporty Group

(Remote)
1 Month ago
Trend Micro - Fullstack Development Engineer

Trend Micro

Manila, Metro Manila, Philippines (On-Site)
16 Years ago
Luxoft - Senior React JS Developer

Luxoft

Bengaluru, Karnataka, India (On-Site)
5 Months ago
Nagarro - Staff Engineer, Frontend React

Nagarro

Mumbai, Maharashtra, India (On-Site)
6 Months ago
The Walt Disney Company - Sr Software Engineer (webOS/Tizen)

The Walt Disney Company

Glendale, California, United States (On-Site)
5 Months ago
Welevel - Senior Cloth Artist

Welevel

Munich, Bavaria, Germany (On-Site)
1 Month ago
Nagarro - Associate Staff Engineer

Nagarro

(Remote)
1 Month ago
Next Level Business Services - Tealium Consultant

Next Level Business Services

San Antonio, Texas, United States (On-Site)
6 Months ago

Get notifed when new similar jobs are uploaded

Jobs in Santa Clara, California, United States

Next Level Business Services - Mobile Test Manager

Next Level Business Services

Alpharetta, Georgia, United States (On-Site)
6 Months ago
Unity - Senior Manager, Payroll Americas

Unity

Bellevue, Washington, United States (On-Site)
4 Months ago
Gearbox Software - Biome Artist

Gearbox Software

Frisco, Texas, United States (On-Site)
4 Months ago
Epic Games - Senior Engine Programmer

Epic Games

United States (On-Site)
3 Months ago
Tencent - Senior Games User Researcher

Tencent

Los Angeles, California, United States (On-Site)
3 Months ago
WebFX - Lead Sales Development Representative (SDR) (Open to Remote or On-Site)

WebFX

Harrisburg, Pennsylvania, United States (Remote)
6 Months ago
Meta - Manager, Recruiting Services & Operations

Meta

Austin, Texas, United States (On-Site)
5 Months ago
The Walt Disney Company - Lead Data Scientist

The Walt Disney Company

Burbank, California, United States (On-Site)
5 Months ago
Sinch - Senior Product Designer

Sinch

United States (Remote)
2 Months ago
Next Level Business Services - Java Developer

Next Level Business Services

San Jose, California, United States (On-Site)
5 Months ago

Get notifed when new similar jobs are uploaded

Research & Development Jobs

Cadence - Lead Product Engineer

Cadence

Beijing, Beijing, China (On-Site)
7 Months ago
ByteDance - Tech Lead, Camera Algorithms Engineer

ByteDance

San Jose, California, United States (On-Site)
2 Months ago
Nagarro - Senior Staff Engineer, Machine Learning

Nagarro

Bengaluru, Karnataka, India (On-Site)
3 Months ago
GlobalHunt - Design Engineer

GlobalHunt

Bengaluru, Karnataka, India (On-Site)
7 Months ago
Fabric - Applied Cryptographer, ZKP Research

Fabric

British Columbia, Canada (Remote)
6 Months ago
NVIDIA - Senior ASIC Verification Engineer, Coherent High Speed Interconnect

NVIDIA

Taipei City, Taiwan (On-Site)
3 Months ago
Microsoft - Research Intern - Formal Document Processing

Microsoft

Redmond, Washington, United States (On-Site)
3 Months ago
Cirrus Logic - Summer Intern, Digital Design Engineer

Cirrus Logic

Austin, Texas, United States (On-Site)
6 Months ago
Samsung Semiconductor - Engineer, DRAM Applications

Samsung Semiconductor

Bellevue, Washington, United States (On-Site)
3 Months ago
Luxoft - Design Verification Engineer - Data Fabric Systems

Luxoft

Iași, Iași County, Romania (On-Site)
4 Months ago

Get notifed when new similar jobs are uploaded

About The Company

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.

Santa Clara, California, United States (Hybrid)

Santa Clara, California, United States (On-Site)

Pune, Maharashtra, India (On-Site)

Seoul, South Korea (Hybrid)

Yokne'am Illit, North District, Israel (Hybrid)

Yokne'am Illit, North District, Israel (On-Site)

Tel Aviv-Yafo, Tel Aviv District, Israel (On-Site)

Santa Clara, California, United States (Hybrid)

View All Jobs

Get notified when new jobs are added by NVIDIA

Level Up Your Career in Game Development!

Transform Your Passion into Profession with Our Comprehensive Courses for Aspiring Game Developers.

Job Common Plug