Software Engineer, ML Accelerators

• 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

• 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

Desirable Qualifications

• Experience with inference on edge platforms
• Experience with cloud ML training pipelines
• HPC experience