Staff Software Engineer - Deep Learning Acceleration

Aurora's mission is to deliver the benefits of self-driving technology safely, quickly, and broadly to make transportation safer, increasingly accessible, and more reliable and efficient than ever before. The Aurora Driver will create a new era in mobility and logistics, one that will bring a safer, more efficient, and more accessible future to everyone.

At Aurora, you'll solve massively complex problems alongside other passionate, smart people, growing as an expert while broadening your field of knowledge. For Aurora's latest news, visit aurora.tech or follow us on LinkedIn.

Aurora hires talented people with diverse backgrounds who are ready to help build a transportation ecosystem that will make our roads safer, get crucial goods where they need to go, and make mobility more efficient and accessible for all. 

We're searching for a Software Engineer to focus on Deep Learning Acceleration.

In this role, you will:

Conduct performance analysis and optimization of Deep Learning networks running on the Autonomous Vehicle (AV). The scope of optimization spans to onboard (deployment on the AV) as well as Training on large-scale data-center. SWE will work on various problems like software architecture, system performance and latency. 

Required Qualifications 

• Minimum 5+ years of professional experience 
• BS, MS, or PhD in Computer Science or related field
• Strong programming skills in C++, Python and CUDA
• Experience in high performance computing , parallel programming. 
• Familiarity with at least one deep learning framework (PyTorch, TensorFlow,, etc.)
• Basic understanding of computer vision deep learning models.
• Strong analytical skills, especially for performance troubleshooting (e.g. profiling, roofline model)
• Ability to quickly learn and adapt to new technology
• Ability to work on large code bases and a fast growing environment.
• Strong communication skills
• Comfortable with Linux/other unix environments

Desirable Qualifications

• Programming experience in C++ for a real world robotic system
• Motion planning or other autonomy experience(robotics, systems software, computer vision, gaming, or mapping)
• Experience with TensorRT, OpenAI triton and other inference acceleration tools.

Pay Range 

The base range for this position is $189,000 - $302,000.  Aurora's pay ranges are determined by role, level, and location. Within the range, the successful candidate's starting base pay will be determined based on factors including job-related skills, experience, qualifications, relevant education or training, and market conditions. These ranges may be modified in the future. The successful candidate will also be eligible for an annual bonus, equity compensation, and benefits.

General Description of Benefits and Other Compensation

Aurora offers medical, dental, vision, life/AD&D, and disability insurance.  Aurora also provides a 401k plan, various wellness benefits, family planning benefits, FSA and commuter benefits, and professional development resources.  The successful candidate may take flexible/reasonable vacation, will receive 11 paid company holidays per year, may take reasonable sick time, and is eligible for paid parental leave.   The successful candidate will also receive an equity grant in the form of restricted stock units (subject to vesting requirements), as well as the opportunity to earn an annual bonus.  Relocation assistance may be provided where applicable.  A further description of available benefits can be found here.

#LI-WS1

#Staff

Aurora considers candidates without regard to their race, color, religion, national origin, age, sex, gender, gender identity, gender expression, sexual orientation, marital status, pregnancy status, parent or caregiver status, ancestry, political affiliation, veteran and/or military status, physical or mental disability, or any other status protected by federal or state law. Aurora considers qualified applicants with criminal histories, consistent with applicable federal, state, and local law. We are also committed to providing reasonable accommodations for qualified individuals with disabilities and disabled veterans in our job application procedures. If you need assistance or an accommodation due to a disability, you may contact us at careersiteaccommodations@aurora.tech.

For California applicants, information collected and processed as part of your application and any job applications you choose to submit is subject to Aurora's California Employment Privacy Policy.

To learn more about life at Aurora please visit our company culture