The Senior Research Engineer role at DeepMind focuses on optimizing machine learning models for on-device applications in robotics. The team is dedicated to adapting and enhancing Gemini Robotics models for low-latency environments while ensuring alignment between model architectures and hardware constraints. Engineers will engage with various teams to deliver robust solutions and contribute to the broader research community. This position emphasizes the intersection of model optimization and hardware acceleration.

• Bachelor's degree in Computer Science, Electrical Engineering, or a related field or equivalent practical experience.
• 8 years of experience in optimizing machine learning models for resource-constrained environments.
• Experience in inference for Large Language Models (LLMs), including architectures like Mixture of Experts or diffusion models.
• Excellent programming skills in Python and C++.
• Experience with ML frameworks such as JAX, TensorFlow, or PyTorch, particularly focused on high-performance inference.

• Optimize model performance for on-device use cases in memory, power, and compute constrained environments.
• Influence future Gemini model architectures to match unique robotics use cases.
• Optimize agent and system-level performance, including orchestration of multiple models.
• Drive strong alignment between model architectures and hardware architectures.
• Engage directly with research, software engineering, and hardware engineering teams to deliver end-to-end solutions.

• Employees at Google are often offered benefits like comprehensive health insurance, 401(k) matching, and flexible work arrangements, among other benefits.