Sr Staff Software Engineer, ML Accelerators

Company: Aurora Innovation
Location: Mountain View
Posted on: September 1, 2024

Job Description:

Aurora hires talented people with diverse backgrounds who are ready to help build a transportation ecosystem that will make our roads safer, get crucial goods where they need to go, and make mobility more efficient and accessible for all. We're searching for a Software Engineer to focus on ML Accelerators
In this role, you will

• 
  
• 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
  
  Required Qualifications
  
  • 
    
  • 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
      
      Pay Range:
      The base salary range for this position is $212,000 - $340,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.
      #LI-WS1
      
      #J-18808-Ljbffr

Keywords: Aurora Innovation, Mountain View , Sr Staff Software Engineer, ML Accelerators, IT / Software / Systems , Mountain View, California