Senior ML Accelerator Engineer - GPU
$170k - $258k • Sunnyvale, California, United States of America • Full-time
Posted 5d ago
Remote Work Policy
On-site
Employment Type
Full-time
Categories
Machine Learning Engineer
About the job
The AI Kernels & Compilers team at GM is at the forefront of developing advanced automated driving technologies, aiming for a future with Zero Crashes, Zero Emissions, and Zero Congestion. This role focuses on transforming cutting-edge AI research into production-grade software that runs efficiently and reliably on real vehicles. The team builds high-performance GPU kernels and custom libraries essential for on-vehicle ML inference in ADAS and autonomous driving systems, ensuring core AI workloads are faster, more reliable, and easier to deploy under real-world constraints.
Responsibilities
- Design, implement, benchmark, and iterate on CUDA-based kernels and custom operators for optimal performance on inference workloads.
- Build and improve tooling for profiling, debugging, and validating CUDA kernels and accelerator-backend code.
- Translate model and system requirements into kernel roadmaps, priorities, and project plans in collaboration with AI Solutions, Compilers, and Architecture teams.
- Deliver reusable, reliable, high-performance libraries into production by collaborating with cross-functional teams.
- Maintain high technology standards for GPU kernel development and performance engineering through code reviews.
- Manage relationships with internal customers to ensure kernels and libraries meet real-world needs.
Requirements
- Minimum 2 years of relevant industry experience.
- BS, MS, or PhD in Computer Science or a related technical field.
- Excellent GPU programming skills in CUDA, with a thorough understanding of parallel programming patterns and GPU architecture.
- Hands-on experience benchmarking, profiling, debugging, and optimizing accelerator libraries and kernels using tools like NSight.
- Strong background in software architecture, library design, and design patterns.
- Strong C++ programming skills and comfort with large codebases.
- Solid background in system performance, high-performance computing, and/or architecture-aware optimizations.
- Strong communication and collaboration skills.
Benefits
- Bonus Potential
- Medical, dental, and vision insurance
- 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
- Relocation benefits