Staff Software Engineer - Deep Learning Acceler...
@ Aurora Innovation

Pittsburgh, Pennsylvania, United States
$222,000
On Site
Full Time
Posted 14 hours ago

Your Application Journey

Personalized Resume
Apply
Email Hiring Manager
Interview

Email Hiring Manager

XXXXXXXX XXXXXXXXX XXXXXXXXX***** @aurora.tech
Recommended after applying

Job Details

About Aurora Innovation

Aurora’s mission is to deliver the benefits of self-driving technology safely, quickly, and broadly. Aurora creates a new era in mobility and logistics by making transportation safer, more efficient, and accessible.

Role Overview

As a Staff Software Engineer - Deep Learning Acceleration, you will enhance the performance of Deep Learning networks used in our Autonomous Vehicle (AV) systems. Your work will optimize software architecture, system performance, and latency issues both onboard vehicles and at large-scale data centers.

Key Responsibilities

  • Conduct performance analysis and optimization of deep learning networks for AV systems.
  • Optimize software architecture, system performance, and latency.
  • Deploy deep learning models on vehicles and during large-scale training.
  • Troubleshoot issues using profiling and roofline model techniques.
  • Collaborate with cross-functional teams to enhance self-driving technology efficiency.

Required Qualifications

  • 5+ years of professional software engineering experience.
  • Degree in Computer Science or related field (BS, MS, or PhD).
  • Strong programming skills in CUDA, C++, and Python.
  • Experience in high-performance computing and parallel programming.
  • Proficiency in performance analysis tools like NVIDIA Nsight Systems and Nsight Compute.
  • Hands-on expertise in deep learning frameworks (e.g., PyTorch, TensorFlow).
  • Strong analytical skills for diagnosing performance bottlenecks.
  • Excellent communication skills for effective teamwork.
  • Comfort with Linux/Unix environments.

Desirable Qualifications

  • Experience in motion planning, robotics, or autonomous systems.
  • Familiarity with TensorRT, OpenAI Triton, Mojo and other acceleration tools.

Compensation & Benefits

The base range for this position is $171,000 - $273,000 with additional bonus, equity, and benefits based on qualifications and market conditions.

Working at Aurora

At Aurora, you will work with talented and experienced individuals in a collaborative, inclusive environment where integrity and teamwork are paramount. Aurora values diversity and provides equal opportunities with a commitment to inclusion.

Key skills/competency

Deep Learning, CUDA, C++, Python, Performance Optimization, Profiling, Autonomous Systems, High-Performance Computing, Linux, Communication

How to Get Hired at Aurora Innovation

🎯 Tips for Getting Hired

  • Research Aurora Innovation's culture: Explore their mission, values, and recent news.
  • Tailor your resume: Highlight deep learning and performance skills.
  • Showcase project impact: Detail relevant optimization experience.
  • Prepare for technical interviews: Practice profiling and GPU optimization challenges.
  • Network strategically: Connect with current employees on LinkedIn.

📝 Interview Preparation Advice

Technical Preparation

Review CUDA and parallel programming basics.
Practice setting up profiling tools like Nsight.
Study roofline model and performance analysis.
Prepare optimization case studies in deep learning.

Behavioral Questions

Describe a challenging project experience.
Explain teamwork in a cross-functional setting.
Discuss handling tight deadlines and pressure.
Share experience adapting to new technologies.

Frequently Asked Questions