Sr Staff Software Engineer, ML Accelerators

Job Description

Posted on: 
December 4, 2023

We’re searching for a Software Engineer to focus on 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

Job 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

Desirable Qualifications

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

Apply now

More job openings