General MotorsRemote
Principal Machine Learning Engineer
United States of America
Full Time
1 day ago
$296,000 - $453,000 USD
H1B Sponsor Likely
Key skills
PythonC++CAIMLPyTorchTensorFlowAWSAzureGCPLeadership
About this role
General Motors is seeking a Principal AI Engineer to lead the design and advancement of their AI platform. This role involves shaping the infrastructure that powers large-scale training and cloud inference, with a focus on distributed training and AI-driven 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
Requirements:
- 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