Staff Research Engineer, LLMs and Generative Media

Google Inc.; Greater London; On-site; GBP 80,000 - 120,000; Full time

Job summary

A leading tech company in Greater London seeks a Staff Research Engineer to lead a new applied machine learning team. The role involves bridging research with real-world applications in areas like generative AI and multi-agent systems. Candidates should have at least 8 years in machine learning, experience leading teams, and expertise in Python and ML frameworks such as JAX, TensorFlow, or PyTorch. The position promises executive-level impact on innovative products while fostering technical excellence among the team members.

Qualifications

• 8 years of experience in applied machine learning.
• Experience building and leading engineering or research teams.
• Experience in Python and ML frameworks like JAX, TensorFlow, or PyTorch.

Responsibilities

• Lead the end-to-end research process.
• Shape and execute the team's technical roadmap.
• Guide research directions and mentor team members.

Skills

Applied machine learning

Leading engineering or research teams

Python

ML frameworks (JAX, TensorFlow, PyTorch)

Published in top-tier ML/AI conferences

Education

Bachelor's degree or equivalent practical experience

Master's degree in Computer Science or a related field

Tools

Python

JAX

TensorFlow

PyTorch

Job description

Staff Research Engineer, LLMs and Generative Media

Google London, UK

Apply

• Bachelor's degree or equivalent practical experience.
• 8 years of experience in applied machine learning, including leading projects from research to production.
• Experience building and leading engineering or research teams.
• Experience in Python and ML frameworks like JAX, TensorFlow, or PyTorch.
• One or more publications in top-tier ML/AI conferences or journals (e.g., NeurIPS, ICML, ICLR, CVPR).

Preferred qualifications:

• Master's degree in Computer Science, a related technical field, or equivalent practical experience.

About the job

The Domain Applied ML (DAML) team is a group within Core ML. Our mission is to accelerate the adoption of AI across Google. We partner with Google Research and DeepMind to translate their breakthroughs (like Gemini) into standardized, efficient solutions. By addressing issues from 0-to-1 prototyping to 1-to-10 scaling, we allow Google’s product areas to innovate with massive quality, performance, and productivity gains.

As a Staff Research Engineer, you will lead a new applied ML team in London, bridging the gap between research from Google Research and DeepMind and its real-world application. You will act as the team's technical anchor, guiding a small group of researchers while dedicating significant time to your own experimentation and coding. Your work will focus on emerging areas like generative AI and multi-agent systems to deliver measurable impact on products like Search, YouTube, and Waymo.

The AI and Infrastructure team is redefining what’s possible. We empower Google customers with breakthrough capabilities and insights by delivering AI and Infrastructure at unparalleled scale, efficiency, reliability and velocity. Our customers include Googlers, Google Cloud customers, and billions of Google users worldwide.

We're the driving force behind Google's groundbreaking innovations, empowering the development of our cutting-edge AI models, delivering unparalleled computing power to global services, and providing the essential platforms that enable developers to build the future. From software to hardware our teams are shaping the future of world-leading hyperscale computing, with key teams working on the development of our TPUs, Vertex AI for Google Cloud, Google Global Networking, Data Center operations, systems research, and much more.

Responsibilities

• Act as the technical expert for a small team. Guide research directions and mentor team members through your own direct contributions.
• Lead the end-to-end research process, from defining novel problems and prototyping solutions to publishing results and partnering with product teams to ship features.
• Shape and execute the team's technical roadmap by collaborating with stakeholders in Google Research, DeepMind, and product areas to identify and pursue opportunities.
• Utilize and advance techniques on Google's infrastructure. Your work will involve Python, JAX/TensorFlow, and focus on areas like Parameter-Efficient Tuning (PET), multimodal modeling, and agentic systems using models like Gemini.
• Research and product by translating technical concepts into actionable plans and user value, while advocating research excellence through publications in conferences and open-source contributions.

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 .

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.