Sr Staff Software Engineer – ML Accelerators
| Company | Aurora Innovation |
|---|---|
| Location | Pittsburgh, PA, USA |
| Salary | $191000 – $306000 |
| Type | Full-Time |
| Degrees | Bachelor’s, Master’s, PhD |
| Experience Level | Senior |
Requirements
- Minimum 5+ years of professional experience
- BS, MS, or PhD in Computer Science or related field
- Strong programming skills in C++, Python
- Familiarity with at least one deep learning framework (PyTorch, TensorFlow, MXNet, etc.)
- Basic understanding of computer vision deep learning models
- Strong analytical skills, especially for performance troubleshooting (e.g. profiling, roofline model)
- Familiarity with CUDA, OpenCL, OpenVX, Halide or other SIMD programming models
- Familiarity with ML-specific accelerators (NPUs, TPUs, IPUs, etc…)
- 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
Responsibilities
- Work closely with our autonomy and hardware teams to understand our on-vehicle ML technology
- Develop perspectives on where opportunities and gaps might be in our machine learning software and hardware strategy with a longer-term horizon in mind
- Maintain relationships and intel across the ML ecosystem including both established players and start-ups (accelerators, software modules, etc.) for both existing and future products
- Execute internal research and development of ML software and hardware technology
- Surface high-impact findings to relevant Engineering leadership, keeping feedback loop going to influence Aurora’s ML strategy
- Help drive value to key engineering stakeholders
Preferred Qualifications
- Experience with inference on edge platforms
- Experience with cloud ML training pipelines
- HPC experience