AI Research Engineer - Diffusion Models & Collaborations

Google LLC; Mountain View (CA); On-site; USD 174,000 - 252,000; Full time

Job summary

Google LLC is seeking a research-focused Software Engineer embedded within our AI and DeepMind ecosystem to drive large-scale experiments and deploy breakthrough ideas across projects from conception to implementation. The role emphasizes collaboration with internal teams, universities, and external labs to advance AI tech and scalable systems.

Candidates will contribute to experiments, new architectures, and publications while aligning with safety and ethics priorities, and will engage with

Qualifications

• Bachelor's degree in Computer Science, Mathematics, Applied Statistics, Machine Learning or equivalent practical experience.
• 5 years of experience with TensorFlow or similar ML frameworks (e.g., JAX or PyTorch).
• 2 years of experience in an applied research setting.

Responsibilities

• Lead ambitious team collaborations, aligning efforts across DeepMind to execute on shared, high-impact research goals.
• Report and present complex research breakthroughs clearly and efficiently.
• Build, lead, and maintain strategic collaborations with key external research labs and leading individuals in the field.
• Act as a principal representative of our work internally and externally at global venues.

Skills

TensorFlow experience

Applied research experience

Training diffusion models

Industry experience from PoC to prod

Education

Bachelor's degree or equivalent in CS/Math/Stats/ML

Job description

Google LLC is seeking a research-focused Software Engineer embedded within our AI and DeepMind ecosystem to drive large-scale experiments and deploy breakthrough ideas across projects from conception to implementation. The role emphasizes collaboration with internal teams, universities, and external labs to advance AI tech and scalable systems.

Candidates will contribute to experiments, new architectures, and publications while aligning with safety and ethics priorities, and will engage with