April 30
🏢 In-office - Bay Area
• Report into the TLM of ML performance • Profile and debug model performance bottlenecks • Improve model and data ingestion operations • Implement novel custom kernels to improve performance and cost • Suggest alternative modeling architectures • Design and implement benchmarks for performance optimization • Work across Planner, Research, and Evaluation to improve simulation and on-board performance
• Advanced degree in Computer Science, Machine Learning, Robotics, or similar technical field • Experience with popular ML frameworks such as JAX, PyTorch, or TensorFlow • Code and testing skills in Python • Experience with GPU programming/optimization using CUDA or similar technologies • Prior experience optimizing ML model performance • Experience with model optimization work deployed in industrial settings • Low-level TensorFlow experience related to performance • Code and testing skills in C++
• Top-notch medical, dental and vision insurance • Mental wellness support • Flexible Spending Account (FSA) • Health Saving Account (HSA) • Special wellness programs • Competitive compensation • Bonus opportunities • Equity • Generous 401(k) plan • Financial coaching • 529 College Savings Plan • Lots of other perks and employee discounts • Flexible work locations • On-site or hybrid work model • Paid time off • Bereavement leave • Sick leave • Parental leave • Enhanced leave options for growing families • Fertility care and adoption support • Education reimbursement • Personal and professional development • Mentorship programs • Employee Resource Groups (ERGs) • Other internal groups • Volunteer time off • Access to Google offices, cafes, wellness centers, massages • At-home fitness and cooking classes
Apply Now