Note: The job is a remote job and is open to candidates in USA. General Motors is an automotive company focused on advancing safer, more sustainable, and more equitable transportation. The Principal Machine Learning Engineer will lead the design and advancement of AI platform infrastructure for large-scale model training and cloud inference, while driving distributed training efficiency, platform scalability, and AI capabilities for autonomous driving systems.
Responsibilities
- Architect, build, and optimize core AI/ML platform infrastructure to support massive-scale model training
- Collaborate with data scientists, ML engineers, and software developers to enable seamless workflows from research to production
- Drive efficiency in large-scale distributed training and data processing pipelines
- Establish best practices for reliability, scalability, and performance across the AI/ML platform
- Provide technical leadership and mentorship, guiding teams on platform design, architecture decisions, and emerging technologies
- Partner with cross-functional stakeholders to align platform capabilities with business needs and strategic AI initiatives
Skills
- * Bachelor's degree or higher in Computer Science, related field, or equivalent experience
- * 8+ years of professional software engineering experience
- * 4+ years of specialized experience in AI/ML domain (e.g., enabling distributed training for large-scale models)
- * Strong programming skills in Python, with proficiency in frameworks such as PyTorch (preferred) or TensorFlow
- * Experience with distributed systems, GPU computing, and cloud environments (AWS, GCP, or Azure)
- * Comfortable operating in highly ambiguous and dynamic environments
- * Willingness to travel to Sunnyvale, CA as needed
- * Proven track record of self-motivation, execution, and delivering impact
- * Deep expertise with PyTorch 2.x+ and distributed training frameworks
- * Strong skills in profiling, analysis, debugging, and optimizing training performance (e.g., avoiding memory fragmentation, operation fusion)
- * Proficiency in C++ for performance-critical components
- * Experience leading cross-functional projects and aligning diverse stakeholders on priorities
Benefits
- Remote work, with employees within a 50-mile radius of Austin, Detroit, Warren, Milford or Mountain View expected to report to that location three times a week, at minimum.
- An incentive pay program offers payouts based on company performance, job level, and individual performance.
- Medical benefits
- Dental benefits
- Vision benefits
- Health Savings Account
- Flexible Spending Accounts
- Retirement savings plan
- Sickness and accident benefits
- Life insurance
- Paid vacation and holidays
- Tuition assistance programs
- Employee assistance program
- GM vehicle discounts
Company Overview
Company H1B Sponsorship