Machine Learning Engineer in United State

About The Company

General Motors (GM) is a global leader in automotive manufacturing and innovation, committed to shaping the future of transportation. With a rich history of technological advancements and a strong focus on sustainability, GM strives to deliver innovative mobility solutions that enhance safety, reduce environmental impact, and improve the driving experience. The company's vision encompasses a world with Zero Crashes, Zero Emissions, and Zero Congestion, reflecting its dedication to creating safer, cleaner, and more efficient transportation options for all. GM fosters a culture of inclusivity, collaboration, and continuous improvement, encouraging employees to contribute to meaningful change and technological breakthroughs that redefine mobility.

About The Role

We are seeking an experienced and technically proficient Staff ML System Engineer to join our team remotely, with the expectation of reporting to our Mountain View location at least three times a week if residing within a 50-mile radius. In this pivotal role, you will be responsible for designing, developing, and optimizing scalable, reliable, high-performance machine learning training infrastructure that supports cutting-edge AI research and model development initiatives. You will work closely with machine learning engineers, research scientists, and cross-functional teams to build robust AI platforms that enable advanced autonomous driving technologies across GM's vehicle lineup. Your expertise will directly impact the efficiency, scalability, and operational excellence of our AI training workflows, ensuring that GM remains at the forefront of automotive AI innovation.

Qualifications

• Bachelor's degree or higher in Computer Science, Electrical Engineering, or a related field, or equivalent professional experience.
• Minimum of 7+ years of professional software engineering experience.
• At least 3+ years of specialized experience in AI/ML infrastructure, including enabling distributed training for large-scale models.
• Proficiency in Python programming, with hands-on experience using frameworks such as PyTorch (preferred), TensorFlow, or similar.
• Experience with distributed computing, GPU computing, and cloud environments such as AWS, GCP, or Azure.
• Willingness to travel to Sunnyvale, CA, as needed.
• Ability to work effectively in highly ambiguous and dynamic environments.

Responsibilities

• Lead the design and development of scalable, reliable machine learning frameworks to support large-scale model training.
• Conduct performance analysis and optimization of training workflows to maximize resource utilization across heterogeneous hardware environments, reducing costs and improving efficiency.
• Enhance system observability, debuggability, and operational excellence to ensure platform stability and user satisfaction.
• Collaborate with cross-functional teams to integrate new features, tools, and technologies into the existing ML platform infrastructure.
• Develop and implement solutions supporting advanced training techniques such as Fully Sharded Data Parallel (FSDP), Pipeline Parallelism, and other scalable training solutions for large foundational models.
• Profile, analyze, debug, and optimize training performance, including data loading and processing pipelines.
• Provide technical leadership and mentorship to team members, fostering a culture of continuous improvement and innovation.

Benefits

• Comprehensive health and wellbeing programs including medical, dental, and vision coverage.
• Health Savings Account (HSA) and Flexible Spending Accounts (FSA).
• Retirement savings plan with company contributions.
• Sickness and accident benefits, life insurance, and paid vacation and holidays.
• Tuition assistance programs to support ongoing education and professional development.
• Employee assistance program promoting mental health and wellness.
• GM vehicle discounts and other exclusive employee perks.
• Potential relocation benefits for eligible candidates.

Equal Opportunity

General Motors is committed to creating a diverse and inclusive workplace where all employees are valued and respected. We provide equal employment opportunities regardless of sex, race, color, national origin, citizenship status, religion, age, disability, pregnancy or maternity status, sexual orientation, gender identity, veteran status, or any other protected characteristic. We believe that fostering an inclusive environment enhances innovation and drives better products and services for our customers.