Senior GPU Engineer [Remote Friendly]

Key Skills

Aartificial intelligence, Analytical and Problem solving, API, C++, CUDA kernels, CUDA/GPU programming, Design, Effective communication skills, Machine learning techniques, OpenCL, Optimization

Job Description

We're Cruise, a self-driving service designed for the cities we love.

We’re building the world’s most advanced self-driving vehicles to safely connect people to the places, things, and experiences they care about. We believe self-driving vehicles will help save lives, reshape cities, give back time in transit, and restore freedom of movement for many.

In our cars, you’re free to be yourself. It’s the same here at Cruise. We’re creating a culture that values the experiences and contributions of all of the unique individuals who collectively make up Cruise, so that every employee can do their best work. 

Cruise is committed to building a diverse, equitable, and inclusive environment, both in our workplace and in our products. If you are looking to play a part in making a positive impact in the world by advancing the revolutionary work of self-driving cars, come join us. Even if you might not meet every requirement, we strongly encourage you to apply. You might just be the right candidate for us.

About the Role:

The Autonomous Vehicle (AV) software stack is a sophisticated workload that includes compute and data intensive functions within perception, prediction, planning and control. This role is within the ML Accelerators (MLA) team which is responsible for maintaining and optimizing the software stack for real-time, on-road performance. This means efficient workload mapping onto a sophisticated, resource-constrained hardware, co-architecting the HW platform with HW engineers and building platforms and tools for AI deployment and performance profiling, diagnosis and optimizations that can be employed by AV engineers (from perception, prediction, planning and controls). MLA engineers work in close collaboration with AV engineers throughout the lifecycle of crafting new AV features which is crucial to the success of our mission.

As a Senior Software Engineer (in Cruise’s Machine Learning Accelerator’s (MLA) Kernels and Libraries team), you will be responsible for the development of high-performance, reliable accelerator solutions for Cruise’s next generation custom silicon. Your solutions will be used to implement the core AV functions across ML and non-ML workloads spanning multiple GPU architectures. As Cruise expands our vehicle offerings on our path to productization, your libraries will not only enable Cruise’s low cost custom silicon critical to our long term business needs, but also provide the needed performance portability to ensure our successful scaling.

What you’ll be doing:

• Design, develop, test and deploy GPU solutions that are optimized across multiple GPU architectures (custom accelerator inside AV and GPGPUs in cloud).
• Own the performance of accelerator kernels and libraries (both on and off vehicle).
• Work with cross-functional teams (compiler, performance tooling, runtime, ML framework and application development) to ensure that reusable, reliable high-performance libraries are delivered.
• Review code submissions, provide guidance on design and implementation, break down tasks for junior developers, track execution and efficiently report status.
• Create and implement technology standards, methodologies, processes, and guidelines
• Manage relationships with external vendors and internal customers.

What you must have:

• Excellent GPU programming skills in CUDA or OpenCL with a thorough understanding of parallel programming patterns and GPU architecture preferably across multiple vendors.
• Strong background in software architecture, library design and design patterns.
• Strong C++ programming skills.
• Hands-on experience benchmarking, profiling, debugging and optimizing accelerator libraries and kernels to extract optimal performance.
• Proven background in system performance, high performance computing and architecture-aware optimizations.
• Strong communication, influence, and negotiation skills and the ability to work collaboratively within a team
• Excellent analytical and problem-solving skills
• Minimum 2+ years of relevant industry experience or equivalent experience

Bonus Points!

• MS or PhD in CS, or related technical field or equivalent experience
• Experience with other GPU programming models like SYCL
• Experience with DSP architectures
• Experience with AI and ML frameworks and inference software stack
• Experience with designing software applications for safety critical systems

The salary range for this position is $144,800 - $212,800. Compensation will vary depending on location, job-related knowledge, skills, and experience. You may also be offered a bonus, restricted stock units, and benefits. These ranges are subject to change.