Senior Machine Learning Engineer - Physical AI and Synthetic Data Generation

Nvidia

Actively hiring
Santa Clara, US Posted 18 days ago $224,000$356,500 / year

At a glance

AI generated

TL;DR

Join our Physical AI team as a senior Machine Learning Engineer, where you will develop sophisticated generative pipelines to create high-fidelity synthetic datasets for training advanced AI agents. Your day-to-day involves architecting and implementing multimodal models using transformer, auto-regressive, and diffusion-based architectures to simulate complex physical environments, ensuring precise control over generated data through user-defined parameters. You’ll also establish robust validation processes and automated quality assurance pipelines, leveraging classical computer vision algorithms and VLMs for detailed evaluation. Essential skills include deep expertise in image/video synthesis, proficiency with major DNN libraries like Python, and a strong background in computer graphics or robotics. This role demands experience in assessing synthetic data’s impact on model performance through rigorous validation metrics, alongside familiarity with simulation platforms and 3D sensor modalities.

Skills

Python TensorFlow PyTorch Diffusion_models Multimodal_learning Computer_vision Workflow_management Database_technologies CUDA Simulation_platforms 3D_sensors(Lidar,Radar) CI/CD

What you'll do

  • Develop and implement advanced generative pipelines for high-fidelity synthetic data in Physical AI applications.
  • Build and fine-tune large-scale multimodal models using transformer and diffusion-based architectures.
  • Apply user controls to ensure precise environmental control during data generation.
  • Establish KPI evaluation processes to validate the quality of synthetic datasets.
  • Create automated QA pipelines using classical computer vision algorithms and VLMs.
  • Lead the generation of massive training datasets with state-of-the-art tools and techniques.
  • Optimize ML software performance, test, and document throughout its lifecycle.

What we're looking for

  • BS, MS, or PhD in Computer Science, Robotics, or related field with 12+ years ML experience.
  • Expertise in image/video synthesis using diffusion models and multimodal methods.
  • Proficiency in Python and major DNN libraries for large-scale training datasets.
  • Strong analytical skills to validate synthetic data quality and physical accuracy.
  • Experience in optimizing code efficiency and workflow management for high performance.
  • Collaborative team player with excellent communication skills in a tight-knit environment.

Market check

Salary context

This $224,000–$356,500 range sits above 96% of similar postings on FindRole.

Peer median band

$165,500$241,700

Median floor and ceiling across peers.

Typical midpoint (25–75%)

$168,800$246,150

Middle half of comparable postings.

Based on 240 comparable postings.

* 240 is the maximum number of comparable postings sampled.

Employer

About Nvidia

Nvidia is a leading designer of graphics processing units (GPUs) and system-on-chip units, powering gaming, professional visualization, data centers, and artificial intelligence workloads. Industry: Semiconductors & AI Computing

Nvidia currently has 801 open roles on FindRole.

Listed pay typically runs $184,000–$287,500 across 797 roles with salary data.

Most-posted roles

View all roles at Nvidia

More like this

Similar roles

Senior AI Machine Learning Engineer

The Hartford

US 10 days ago $117,200$175,800
AWS GCP SageMaker Streamlit Python Java C# Hadoop Spark Redshift Snowflake BigQuery Jenkins Terraform GitHub GitHub Actions Apache Airflow Kubernetes Docker SQL CI/CD MLOps

Senior Machine Learning Engineer, AI Platform

Adobe

San Jose, US 17 days ago $211,800$306,625
Python Java C++ Cloud Infrastructure Distributed Computing Deep Learning Virtual Reality Augmented Reality Artificial Intelligence Robotics Interactive Experiences Large-Scale Computing Frameworks Data Analysis Systems Modeling Environments

Senior Machine Learning Engineer – Perception & Embodied AI

General Motors (GM)

Remote (Mountain View Technical Center - Mountain View Technical Center, US) 70 days ago $170,600$261,300
PyTorch TensorFlow Python Computer Vision Deep Learning Data Mining Multi-Modal Sensor Fusion Real-Time Systems Embedded Systems Model Optimization Metric Definition System Integration LiDAR Camera Radar Transformer Models CVPR ECCV ICCV NeurIPS
Remote