Senior AI/ML Performance Engineer
$145k - $261k • Sunnyvale, California, United States of America • Full-time
Posted 25d ago
Job Location
Sunnyvale, California, United States of America
Tech Stack
Remote Work Policy
On-site
Employment Type
Full-time
Categories
AI Infrastructure Engineer
About the job
GM is seeking a Senior Performance Engineer to join the AV Capacity and Performance Engineering team within the AV Infrastructure organization. This role is critical in supporting the development of autonomous vehicles by providing input into large-scale ML infrastructure strategy, advising on cloud budget decisions, identifying and executing optimization projects, and offering capacity planning and engineering expertise. The team's mission is to ensure efficient and scalable ML infrastructure for GM's autonomous vehicle research and development.
Responsibilities
- Adopt and run AV models to support GM's long-term GPU system strategy and infrastructure roadmap.
- Conduct deep-dive analyses of production workloads to identify bottlenecks and propose optimization strategies.
- Collaborate with AI/ML Research, Infrastructure Engineering, and Cloud Vendors on projects to enhance engineering velocity and cost-efficiency.
- Identify opportunities for architectural improvements to ensure scalability and reliability of large-scale ML training and inference environments.
Requirements
- 5+ years of professional experience in high-scale infrastructure or ML systems.
- Bachelor's Degree in Computer Science, a related technical field, or equivalent practical experience.
- Expert-level coding skills in Python.
- Ability to architect and debug within the PyTorch ecosystem.
- Proven track record of resolving performance issues in large-scale distributed production environments.
- Deep understanding of distributed systems, modern ML system design, and high-performance computing (HPC).
- Hands-on experience with Kubernetes for orchestrating complex workloads.
- Technical proficiency with Nvidia DCGM, nvidia-smi, and Grafana for real-time telemetry and observability.
- Extensive experience working within major cloud ecosystems (AWS, GCP, or Azure).
Benefits
- Medical, dental, vision insurance
- Health Savings Account
- Flexible Spending Accounts
- Retirement savings plan
- Sickness and accident benefits
- Life insurance
- Paid vacation & holidays
- Tuition assistance programs
- Employee assistance program
- GM vehicle discounts