Staff Research Engineer, LLMs and Generative Media

Requirements:

• 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).

Nice to haves:

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

What you'll be doing:

• 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.

Perks and benefits:

• Excellent benefits package
• Opportunity to work with cutting-edge technologies
• Collaborative and innovative work environment