Staff Software Engineer, Deep Learning Acceleration

## Responsibilities - Conduct performance analysis and optimization of Deep Learning networks running on the Autonomous Vehicle (AV). - Optimize software architecture, system performance, and latency for deep learning applications. - Work on deployment of deep learning models on the AV and training on large-scale data centers. - Troubleshoot performance issues using profiling and roofline model techniques. - Collaborate with cross-functional teams to enhance the efficiency of self-driving technology. ## Requirements - Minimum 5+ years of professional experience in software engineering. - BS, MS, or PhD 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, Nsight Compute and applying techniques like roofline model for performance optimization. - Hands-on experience in optimizing DL/ML workloads at the framework level using at least one deep learning framework (e.g., PyTorch, 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. ## Nice to Have - Hands-on experience in motion planning or related fields such as robotics, autonomous systems, systems software, or computer vision. - Experience with TensorRT, OpenAI Triton, Mojo and other inference acceleration tools. ## Benefits The base salary range for this position is $171,000 - $247,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.