Research Scientist, Gemini Diffusion, DeepMind

Requirements:

• PhD degree in Computer Science, Electrical Engineering, a scientific discipline, or Mathematics, a related field, or equivalent practical experience.
• 2 years of experience conducting research in machine learning, evidenced by publications or prior research roles.
• Experience in programming with Python-based scientific libraries (e.g., NumPy, SciPy, JAX, or TensorFlow).
• Experience with language models (LLMs), transformers, diffusion models, text diffusion, or distributed training.
• Experience with machine learning, mathematics, and statistics (e.g., linear algebra or calculus).

Nice to haves:

• Experience in C++ or broader programming.
• Experience in data engineering and visualization.
• Experience in large-scale system design and distributed systems.
• Strong communication skills, including experience in technical discussions, presentations, and research writing.
• A track record of building software, either in open source or as part of a company product or research papers.
• Distributed computation for ML, especially in the context of accelerators (e.g., sharding, multi-host computation).

What you'll be doing:

• Drive novel and disruptive research to advance frontier AI models, particularly within text diffusion, by identifying and solving key scientific challenges.
• Prototype and develop new architectures and algorithms, contributing directly to Gemini Diffusion research efforts.
• Validate the theoretical and practical impact of research at scale through experimental design, execution, and in-depth analysis.
• Advance the fundamental architecture, algorithmic design, and capabilities of large-scale diffusion models.
• Collaborate with other Generative AI teams to transition research technologies into production, fostering a culture of deep scientific expertise and accuracy.

Perks and benefits:

• Opportunity to work on cutting-edge AI research projects at Google DeepMind.
• Contribute to advancements in AI development for global challenges and product innovation.
• Chance to publish findings and collaborate with research programs worldwide.
• Diverse learning opportunities and career pathways in a global team setting.