Taro Logo

Software Engineer, ML Accelerator

Autonomous driving technology company developing the Waymo Driver, focused on improving mobility while reducing traffic accidents since 2009.
$170,000 - $216,000
Machine Learning
Senior Software Engineer
Hybrid
1,000 - 5,000 Employees
3+ years of experience
AI · Automotive

Description For Software Engineer, ML Accelerator

Waymo, an industry leader in autonomous driving technology, is seeking a Software Engineer for their ML Accelerator team within the Compute Team. This role is crucial in developing the compute platform that powers Waymo's fully autonomous vehicle software stack. The position offers an exciting opportunity to work on one of the world's highest performance automotive compute platforms.

The role involves designing and implementing full-stack solutions for ML accelerator chips, from firmware to low-level drivers and APIs. You'll be working with cutting-edge technology, optimizing performance for AI workloads, and collaborating closely with hardware engineers throughout the ASIC design and verification processes. The position requires expertise in embedded systems, hardware-software interactions, and strong programming skills in C and C++.

As part of Waymo's Compute Team, you'll be contributing to the development of custom silicon and system-level compute architectures that push the boundaries of performance, power, and latency. The team is multidisciplinary, offering opportunities to work alongside talented professionals in various domains.

The position offers competitive compensation ($170,000-$216,000), comprehensive benefits including medical, dental, and vision insurance, mental wellness support, and equity opportunities. The hybrid work environment provides flexibility, including the option to work from another location for four weeks per year. This is an excellent opportunity for experienced software engineers passionate about autonomous driving technology and hardware-software optimization.

Last updated 20 days ago

Responsibilities For Software Engineer, ML Accelerator

  • Design and implement full stack solution from firmware, low-level drivers, APIs for ML accelerator chips
  • Analyze and optimize firmware and driver performance for demanding AI workloads
  • Collaborate with hardware engineers closely throughout the ASIC design and verification processes
  • Design and implement efficient memory management solutions including NUMA, IOMMU, etc.

Requirements For Software Engineer, ML Accelerator

Python
  • 3+ years of experience in software development
  • Baremetal or RTOS firmware development experience
  • Experience with embedded architectures, IO technologies (e.g. PCIe, DRAM, AXI, etc.), and hardware/software interactions
  • Strong C, C++ skills

Benefits For Software Engineer, ML Accelerator

Medical Insurance
Dental Insurance
Vision Insurance
Mental Health Assistance
Equity
Parental Leave
  • Top-notch medical, dental and vision insurance
  • Mental wellness support
  • Gym membership
  • Special wellness programs
  • Competitive compensation
  • Bonus opportunities
  • Equity
  • Employee provident fund
  • Employee discounts
  • Flexibility to work from another location for four weeks per year
  • Paid time off
  • Bereavement leave
  • Sick leave
  • Parental leave

Jobs Related To Waymo Software Engineer, ML Accelerator