Principal ML Engineer – Embodied AI Scaling Foundations

Jobtailor; California (MO); On-site; USD 150,000 - 210,000; Full time

Job summary

Jobtailor is seeking a highly skilled engineer to scale and align foundation models for autonomous vehicles. You will design data curation, pre-training, and fine-tuning recipes and build end-to-end training pipelines used by distributed teams.

The ideal candidate has advanced degrees and a track record delivering research to production, with PyTorch, Python, and big-data tools. You will mentor juniors and collaborate cross-functionally to elevate onboard driving safety and reliability.

Qualifications

• Bachelor's, Master's, or PhD in CS with robotics/ML focus required.
• Proven experience with large-scale Foundation Models and alignment methods.
• Proven production research experience under tight deadlines.
• Proficiency with PyTorch and Python for model development.
• Experience building scalable, performant training pipelines for distributed teams.
• Strong data processing skills using NumPy, Pandas and Apache Spark.

Responsibilities

• Scale AV foundation model pre-training and fine-tuning with data across sources.
• Design and implement technologies aligned with GM's mission.
• Develop novel data curation and training recipes for AV models.
• Design, architect, and deploy data pipelines for curation and training.
• Apply unsupervised pre-training, imitation learning, RL, guidance and selection.
• Collaborate with cross-functional teams to align distillation recipes for onboard models.
• Conduct research and stay current with AI frameworks and libraries.
• Lead projects from ideation through deployment and document learnings.
• Mentor junior team members and drive continuous improvement.
• Develop ML models that improve safety, reliability, and scalability of autonomous vehicles.

Skills

Foundation Model Development

Data Curation Algorithms

PyTorch Proficiency

Large Model Training Pipelines

Robotics Experience

Education

Bachelor's degree

Master's degree

PhD degree

Tools

Python

NumPy

Pandas

Apache Spark

Jupyter Notebooks

Job description

• Scale AV foundation model pre-training and fine-tuning with data into billions of examples across different sources


• Design and implement technologies aligned with GM's mission


• Develop novel data curation algorithms, pre-training, post-training, and fine-tuning recipes for AV models


• Design, architect, and deploy data curation and training pipelines


• Apply unsupervised pre-training, imitation learning, reinforcement learning, guidance, and selection techniques


• Collaborate with cross-functional teams to integrate and align foundation-model distillation recipes for onboard driving models


• Conduct research and stay current with AI frameworks and libraries


• Lead projects from ideation through deployment and document learnings and best practices


• Lead critical technical initiatives and collaborate with cross-functional teams


• Mentor junior team members and contribute to knowledge sharing and continuous improvement


• Develop ML models that improve the safety, reliability, and scalability of autonomous vehicles

Requirements

• Bachelor's, Master's, or PhD degree in Computer Science with a focus in Robotics and/or Machine Learning, or a related field


• Proven experience working with large-scale Foundation Models and alignment methods


• Proven experience delivering applied research in production environments and maintaining best practices under tight deadlines


• Proficiency with PyTorch and Python


• Proven experience building and scaling large model training pipelines that are performant and enable quick iteration by distributed teams


• Strong data processing skills using NumPy, Pandas, and Apache Spark


• Experience deploying foundation models into production environments and understanding the end-to-end process


• Previous experience in Robotics or Autonomous Driving


• Excellent communication skills to collaborate with diverse teams and stakeholders

Core Competencies

Demonstrates expertise in developing and deploying large-scale Foundation Models, with a strong focus on data curation algorithms and training pipelines for autonomous vehicles. Proficient in collaborating with cross-functional teams and mentoring junior members to drive continuous improvement and innovation.

Highest-signal resume keywords

• Foundation Model Development


• Data Curation Algorithms


• PyTorch Proficiency


• Large Model Training Pipelines


• Robotics Experience

ATS Optimization Keywords

Hard Skills

• Machine Learning


• Data Processing


• Unsupervised Pre-Training


• Imitation Learning


• Reinforcement Learning


• Python Proficiency


• Apache Spark


• NumPy


• Pandas


• Model Deployment

Soft Skills

• Excellent Communication


• Collaboration


• Mentoring

Certifications & Qualifications

• Bachelor's Degree


• Master's Degree


• PhD Degree

Industry Keywords

• Autonomous Vehicles


• AI Frameworks


• Foundation Model Alignment


• Production Environments


• Cross-Functional Collaboration