Senior AI/ML Engineer - Data Scaling, Embodied AI Data Foundations

Train on driving data almost nobody else has - real-world miles from GM's fleets, plus synthetic sim data - scaling into billions of examples. Then decide which ones are worth it: ten thousand near-identical highway miles teach the model less than one unprotected left turn in the rain. Mixture design, curation, mining, and evaluation are how you find out which is which, working alongside other MLEs and research scientists.

Work on questions with no textbook answers yet. Scaling laws for language are well mapped by now; for embodied driving data - heavy-tailed, safety-constrained, closed-loop - they aren't. You'd be helping write them, and we support publishing what you find.

See your results in the world rather than on a leaderboard. The models this team ships change how the vehicle behaves on real roads, and that behavior comes back as the evidence for your next iteration.

As a Senior AI/ML Engineer in the Embodied AI Data Foundations organization, you will be an individual contributor developing data-centric AI solutions that directly improve autonomous driving performance. You will design and run the data curation and model training recipes that produce models capable of safe, reliable behavior across diverse real-world scenarios, drawing on both real and synthetic data.

What You'll Do

Design and run experiments that connect data composition to model behavior: dataset mixtures, sampling strategies, curricula, and scaling-law studies that tell us where to invest next.

Apply methods such as self-supervised pre-training, imitation learning, reinforcement learning, and foundation-model fine-tuning to driving behavior, trajectory generation, and perception tasks.

Develop data curation and mining methods - auto-labeling, deduplication, difficulty and uncertainty estimation, long-tail and out-of-distribution scenario discovery - to raise the value of every training example.

Define offline metrics and evaluations that actually predict on-road behavior, and use them to make model and data decisions from evidence rather than intuition.

Trace model failures back to their root cause in the data, then close the loop by specifying the data needed to fix them.

Train models at scale across large multi-GPU/multi-node datasets, partnering with platform teams on the pipelines and tooling this requires.

Collaborate with cross-functional teams to bring models into onboard driving systems, and document learnings and best practices along the way.

Follow relevant literature and bring promising advances into our recipes and evaluations.

Your Skills and Abilities (Required Qualifications)

Master's or PhD in Computer Science, Robotics, Machine Learning.

Strong ML fundamentals: you can design a clean experiment, pick the right baseline, read an ablation, and tell signal from noise.

Proficiency in Python and PyTorch, with experience training models on large datasets.

Hands-on experience with data-centric ML: curation, sampling, labeling, or evaluation of large training sets.

Working knowledge of large-scale foundation models and how they are pre-trained, fine-tuned, and aligned.

Solid data analysis skills (NumPy, Pandas; SQL or Spark for large datasets).

Demonstrated ability to deliver applied ML results under real-world constraints and timelines.

Clear communication: you can explain a result and its limits to both engineers and non-experts.

Preferred:

PhD, publications, or open-source contributions in representation learning, multimodal or vision-language models, generative models, RL, or data-centric ML.

Experience with robotics, autonomous driving, or other embodied AI systems.

Experience with synthetic and simulation data, including sim-to-real transfer.

Familiarity with production ML deployment workflows.

Compensation:

The salary range for this role is $125,000 to $174,500. The actual base salary a successful candidate will be offered within this range will vary based on factors relevant to the position.

GM DOES NOT PROVIDE IMMIGRATION-RELATED SPONSORSHIP FOR THIS ROLE. DO NOT APPLY FOR THIS ROLE IF YOU WILL NEED GM IMMIGRATION SPONSORSHIP NOW OR IN THE FUTURE