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 improve self-driving technology.
Key Responsibilities
- Conduct performance analysis and optimization of Deep Learning networks used in 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 such as 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
Compensation/salary range: $171,000 - $273,000
Annual bonus
Equity compensation
Benefits
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 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 Pittsburgh, Pennsylvania
Patient Care Consultant
Jushi
Pittsburgh
Full Time
Nov 3
Patient Care Consultant
Jushi
Pittsburgh
Full Time
Nov 17
Shift Supervisor
Jushi
Pittsburgh
Full Time
Nov 18