Software Engineer, Machine Learning/AI Accelerator
Core
Design and implement full-stack firmware and driver solutions for ML accelerator chips to support Waymo's autonomous vehicle compute platform.
Role type
Senior embedded software engineer (ML accelerator firmware)
Builds
High-performance custom silicon and system-level compute architectures for autonomous driving
Domain
Autonomous vehicles / Embedded systems / Hardware-software co-design
Deliverable
production ML models
Required skills
Baremetal or RTOS firmware development, Embedded architectures (PCIe, DRAM, AXI), C, C++, Memory management (NUMA, IOMMU)
Preferred skills
RISC-V architecture, ML frameworks (TensorFlow, JAX), Silicon emulator, Hardware-software co-design principles
Technologies
TensorFlow, JAX, PCIe, DRAM, AXI, NUMA, IOMMU, RISC-V
Responsibilities
Design and implement full stack solution from firmware, low-level drivers, and APIs for ML accelerator chips; Analyze and optimize firmware and driver performance for demanding AI workloads; Collaborate with hardware engineers throughout ASIC design and verification processes; Design and implement efficient memory management solutions including NUMA and IOMMU
Seniority
Mid-level, hands-on IC
