Staff ML Engineer
Job Summary
As a Staff AI/ML Engineer in the Embodied AI Data Foundations organization, you will serve as a senior individual contributor developing machine learning solutions, maximally leveraging multiple sources of data including real and synthetic datasets, that directly impact autonomous driving performance. You will design and implement advanced data curation and model training recipes that will deliver ML models enabling safe and reliable vehicle behavior across diverse real-world scenarios. In this role, you will partner closely with cross-functional engineering teams, contribute to core technical direction within your domain, and support the growth of engineers through technical collaboration and mentorship.
Responsibilities
- Design and implement ML solutions for autonomous driving
- Apply unsupervised pre-training, imitation, and reinforcement learning
- Build efficient infrastructure and data pipelines for model iterations
- Mentor and support engineering team members through collaboration
Required Skills
- Proficiency in PyTorch and Python
- Experience with large-scale foundation models
- Strong data processing skills (NumPy, Pandas, Spark)
- Background in robotics or autonomous driving systems
Job Details
What You'll Do
- Design and implement ML solutions aligned with GM's autonomous driving objectives.
- Apply techniques such as unsupervised pre-training, imitation learning, reinforcement learning, model scaling/selection, foundation modeling, to solve problems in object detection/tracking/classification, trajectory generation, and safe AI.
- Collaborate with cross-functional teams to deploy models and algorithms into onboard driving systems.
- Contribute to applied research efforts and remain current with advancements in ML frameworks and methods.
- Design and build efficient infrastructure, pipelines, and tooling to facilitate fast-pace model iterations.
- Drive technical execution from prototyping through production deployment, documenting learnings and best practices.
- Support and mentor engineers through technical collaboration and code reviews, fostering knowledge sharing and engineering excellence.
Your Skills & Abilities
- Bachelor's, Master's, or PhD in Computer Science, Robotics, Machine Learning, or related field.
- Experience working with large-scale foundation models and alignment methods applied to real-world systems.
- Demonstrated ability to deliver applied ML solutions under real-world constraints and timelines.
- Proficiency in PyTorch and Python.
- Experience building and scaling model training pipelines enabling efficient iteration across teams.
- Strong data processing skills using tools such as NumPy, Pandas, and Apache Spark.
- Strong communication skills enabling effective collaboration across engineering teams.
- Experience deploying ML models into production environments and understanding end-to-end deployment workflows.
- Experience in robotics or autonomous driving systems preferred.