Research Engineer, Deployment Performance, Robotics, DeepMind

Requirements:

• Bachelor's degree in Computer Science, Mathematics, a related field, or equivalent practical experience.
• 5 years of experience with machine learning frameworks (e.g., JAX, PyTorch).
• 2 years of experience in Python.
• 2 years of experience in optimizing for model performance, with a focus on inference on TPUs.
• 2 years of experience in building, profiling, and optimizing systems.
• 1 year of experience in C++.

Nice to haves:

• 1 year of experience with related infrastructure (e.g., Servo, One Platform, Gemax, Scaffolding, Uniserve, etc.).
• 1 year of experience with robotics or control systems.

What you'll be doing:

• Optimize and maintain serving setups for various environments, including research, on-premise, and on-device deployment.
• Conduct performance analysis, profiling, and optimization for serving robotics models.
• Collaborate effectively with a large team to meet ambitious research goals in a fast-moving environment.

Perks and Benefits:

Information collected and processed as part of your Google Careers profile, and any job applications you choose to submit is subject to Google's Applicant and Candidate Privacy Policy.

Google is proud to be an equal opportunity and affirmative action employer. We are committed to building a workforce that is representative of the users we serve, creating a culture of belonging, and providing an equal employment opportunity regardless of race, creed, color, religion, gender, sexual orientation, gender identity/expression, national origin, disability, age, genetic information, veteran status, marital status, pregnancy or related condition (including breastfeeding), expecting or parents-to-be, criminal histories consistent with legal requirements, or any other basis protected by law. See also Google's EEO Policy, Know your rights: workplace discrimination is illegal, Belonging at Google, and How we hire.

If you have a need that requires accommodation, please let us know by completing our Accommodations for Applicants form.

Google is a global company and, in order to facilitate efficient collaboration and communication globally, English proficiency is a requirement for all roles unless stated otherwise in the job posting.

To all recruitment agencies: Google does not accept agency resumes. Please do not forward resumes to our jobs alias, Google employees, or any other organization location. Google is not responsible for any fees related to unsolicited resumes.