Jobs for Developers

Staff Accelerator Engineer - Compute Kernels [Remote Friendly]

CruiseSan Francisco, CaliforniaFull-time$193kOct 20, 2023
Apply for this job

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.

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 essential to the success of our mission.

As a Staff Software Engineer in Cruise’s Machine Learning Accelerator’s (MLA) Kernels and Libraries team, you will be responsible for the design and development of high-performance, safety-critical reliable accelerator solutions for Cruise’s AV workloads across our compute platforms. 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:

  • Architect, design and develop accelerator libraries that are optimized across multiple architectures (custom accelerator inside AV and GPGPUs in the cloud).
  • Build optimized ML inference primitives and custom application workloads that support multiple accelerators.
  • Collaborate with compiler and runtime teams to build scalable interfaces for core primitive compute libraries, ML teams to guide the architecture of models and application teams to design the right algorithms that make efficient use of compute resources.
  • Own the performance of accelerator kernels and libraries used throughout the AV stack.
  • Create and implement coding standards, methodologies, processes, and guidelines. Coach and mentor team members on engineering excellence projects.
  • Partner with external vendors to deliver improvements and internal customers to drive requirements.

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 with the ability to feel comfortable in large codebases.
  • Hands-on experience benchmarking, profiling, debugging and optimizing accelerator libraries and kernels to extract optimal performance.
  • Solid 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.
  • Exceptional leadership skills and ability to set direction and work through ambiguities
  • Minimum 5+ years of relevant industry experience or equivalent experience

Bonus Points!

  • MS or PhD in CS, or related technical field or equivalent experience
  • Experience with GPU programming frameworks like SYCL, Triton, CUB, Thrust, CUTLASS and AITemplate
  • Experience with ML frameworks (e.g. Pytorch, JAX, Tensorflow)
  • Experience with ARM NEON, SVE, AVS or SSE-style SIMD intrinsics programming and libraries like Eigen, Highway and XNNPACK
  • Experience with DSP programming

The salary range for this position is $193,900 - 285,000. 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.

Why Cruise?

  • Our benefits are here to support the whole you:

    • Competitive salary and benefits 
    • 401(k) Cruise matching program 
    • Medical / dental / vision, AD+D and Life
    • Subsidized mental health benefits
    • Flexible vacation and company paid holidays
    • Healthy meals and snacks available for non-remote employees
    • Paid parental, jury duty, bereavement, family care, and medical leave
    • Fertility Benefits
    • Dependent Care Flexible Spending Account
    • Flexible Spending Account 
    • Pre-tax Commuter Benefit Plan for non-remote employees
    • CruiseFlex, a working policy for US-Based Cruisers, lets you and your manager find the working style that’s best for you, whether it’s primarily in-person, primarily at home, or a combination of home and in-office time. - learn more about CruiseFlex here
  • We’re Integrated

    • Through our partnerships with General Motors and Honda, we are the only self-driving company with fully integrated manufacturing at scale.

  • We’re Funded

    • GM, Honda, Microsoft, T. Rowe Price, and Walmart have invested billions in Cruise. Their backing for our technology demonstrates their confidence in our progress, team, and vision and makes us one of the leading autonomous vehicle organizations in the industry. Our deep resources greatly accelerate our operating speed.
  • We’re Independent

    • We have our own governance, board of directors, equity, and investors. Our independence allows us to not just work on the edge of technology, but also define it.
  • We’re Vested

    • You won’t just own your work here, you’ll have the potential to own equity in Cruise, too. We are competing in a market that is projected to grow exponentially, which gives our company valuation room to grow. We offer a new kind of equity program called Recurring Liquidity Opportunity (RLO), which combines IPO-like liquidity with the stability of remaining private - learn more about RLO here
  • We’re Safety Conscious

    • We integrate #staysafe, our top priority at Cruise, into our everyday work. Through our Safety Management System, every Cruiser is asked to do their part by reporting any potential issues or hazards they observe and making continuous improvements. You’ll be able to contribute to safety at Cruise, no matter your job function or title.

Cruise LLC is an equal opportunity employer. We strive to create a supportive and inclusive workplace where contributions are valued and celebrated, and our employees thrive by being themselves and are inspired to do the best work of their lives. We seek applicants of all backgrounds and identities, across race, color, caste, ethnicity, national origin or ancestry, citizenship, religion, sex, sexual orientation, gender identity or expression, veteran status, marital status, pregnancy or parental status, or disability. Applicants will not be discriminated against based on these or other protected categories or social identities. Cruise will consider for employment qualified applicants with arrest and conviction records, in accordance with applicable laws.

Cruise is committed to the full inclusion of all applicants. If reasonable accommodation is needed to participate in the job application or interview process please let our recruiting team know or email [email protected].

We proactively work to design hiring processes that promote equity and inclusion while mitigating bias. To help us track the effectiveness and inclusivity of our recruiting efforts, please consider answering the following demographic questions. Answering these questions is entirely voluntary. Your answers to these questions will not be shared with the hiring decision makers and will not impact the hiring decision in any way. Instead, Cruise will use this information not only to comply with any government reporting obligations but also to track our progress toward meeting our diversity, equity, inclusion, and belonging objectives.

Candidates applying for roles that operate and remotely operate the AV: Licensed to drive a motor vehicle in the U.S. for the three years immediately preceding your application, currently holding an active in-state regular driver’s license or equivalent, and no more than one point on driving record. A successful completion of a background check, drug screen and DMV Motor Vehicle Record check is also required.

Note to Recruitment Agencies: Cruise does not accept unsolicited agency resumes. Furthermore, Cruise does not pay placement fees for candidates submitted by any agency other than its approved partners. 

Alternative Jobs