Job Description
The Staff Software Engineer focusing on Deep Learning Acceleration at Aurora is responsible for analyzing and optimizing the performance of deep learning networks used in autonomous vehicle systems, ensuring efficient deployment and training, and collaborating with cross-functional teams to enhance self-driving technology.
Key Responsibilities
- Conduct performance analysis and optimization of Deep Learning networks for autonomous vehicle systems.
- Optimize software architecture, system performance, and latency for deep learning applications.
- Deploy deep learning models on autonomous vehicles and train models on large-scale data centers.
- Troubleshoot performance issues using profiling tools and techniques like the roofline model.
- Collaborate with cross-functional teams to improve the efficiency and reliability of self-driving technology.
Requirements
- Minimum 5 years of professional experience in software engineering.
- Bachelor's, Master's, or PhD degree in Computer Science or a related field.
- Strong programming skills in CUDA, C, and Python.
- Extensive experience in high-performance computing and parallel programming, specializing in optimizing workloads to reduce GPU memory usage, minimize latency, and/or maximize throughput.
- Proficiency in leveraging performance analysis tools such as NVIDIA Nsight Systems and Nsight Compute, and applying techniques like the roofline model for performance optimization.
- Hands-on experience in optimizing Deep Learning and Machine Learning workloads at the framework level using at least one deep learning framework such as PyTorch or TensorFlow, ensuring efficient and scalable model deployment.
- Strong understanding of the fundamentals of computer vision and transformer-based deep learning architectures, with proficiency in foundational neural network building blocks.
- Strong analytical skills for diagnosing and troubleshooting performance bottlenecks in complex systems.
- Demonstrated ability to quickly learn and adapt to emerging technologies and tools in a fast-paced environment.
- Experience working on large code bases in a fast-growing environment.
- Strong communication skills, enabling effective teamwork across multidisciplinary teams.
- Comfortable working in Linux/Unix environments.
Benefits & Perks
Salary range: 189,000 - 303,000 USD
Annual bonus
Equity compensation
Benefits (unspecified)
Ready to Apply?
Join Aurora and make an impact in renewable energy
Stay Updated on Sustainability Jobs
Get the latest renewable energy jobs and career tips delivered to your inbox.
Job Alerts
Get notified about new sustainability jobs
More at Aurora
More jobs at Aurora
Data Platform Engineer
Aurora
NEW
Pittsburgh
Full Time
13h
$105k-157k
Security Engineering Technical Lead Manager TLM - Aurora Enterprise Security
Aurora
NEW
Seattle
Full Time
2d
$189k-274k
FP A Analyst Procurement Systems
Aurora
Pittsburgh
Full Time
3d
$104k-166k
More jobs in Mountain View, California
Senior Software Engineer, Machine Learning, Calibration
Nuro
Mountain View
Full Time
Dec 17
$194k-291k
Senior Staff Software Engineer, Behavior Verification
Nuro
Mountain View
Full Time
Dec 22
$176k-320k
Senior Staff Software Engineer Localization Visual Odometry
Gatik
Mountain View
Full Time
Dec 25
$170k-260k