Senior Staff Machine Learning Engineer, LLM/VLM Model Architecture & Optimization

📋 Description Design VLM/LLM model architecture and align with hardware architectures. Optimize model performance for on-device use cases (memory, power, compute constrained Collaborate with research, software, hardware, and product teams to deliver end-to-end solutions. 🎯 Requirements 7+ years in Machine Learning with large-scale model development (LLM/VLM or foundation models). Expertise in low-latency on-device inference and hardware acceleration. Extensive experience with PyTorch, JAX and large-scale model training. Ability to operate under ambiguity and set direction amid evolving constraints. Experience applying LLMs or foundation models in high-reliability domains (autonomy/robotics). Master's in CS, EE, or related field, or equivalent practical experience. 🎁 Benefits Waymo employees are eligible for discretionary annual bonus, equity incentives, and company Salary range listed below with potential location-based adjustments. Access to data from millions of miles of driving and multi-sensor data for research and development. Opportunities to publish in top ML/vision venues and collaborate with cross-disciplinary teams. Career growth in a leading autonomous driving company.