Research Engineer, AI for Weather and Energy

Research Engineer, AI for Weather and Energy

San Francisco

At Google DeepMind, we value diversity of experience, knowledge, backgrounds and perspectives and harness these qualities to create extraordinary impact. We are committed to equal employment opportunities regardless of sex, race, religion or belief, ethnic or national origin, disability, age, citizenship, marital, domestic or civil partnership status, sexual orientation, gender identity, pregnancy, or related condition (including breastfeeding) or any other basis as protected by applicable law. If you have a disability or additional need that requires accommodation, please do not hesitate to let us know.

Snapshot

At Google DeepMind, we've built a unique culture and work environment where long-term ambitious research can flourish. Our team is part of our Sustainability Program, whose aim is to revolutionize environmental sustainability with AI. We conduct fundamental research which develops novel AI methods, and we translate our research advances into real-world applications and products. Notably, our team pioneers AI weather models, including award-winning GraphCast, GenCast and FGN, which are revolutionizing weather forecasting. We prototype research ideas, scale them and deploy our models in real-time.  This role focuses on developing cutting-edge deep learning and engineering methods, and applying them to sustainability challenges, with opportunities to work on problems related to weather, climate, energy, and other key domains.

About us

Artificial Intelligence could be one of humanity’s most useful inventions. At Google DeepMind, we’re a team of scientists, engineers, machine learning experts and more, working together to advance the state of the art in artificial intelligence. We use our technologies for widespread public benefit and scientific discovery, and collaborate with others on critical challenges, ensuring safety and ethics are the highest priority.

The role

Research Engineers at Google DeepMind lead our efforts in developing and scaling novel algorithmic methods that push the frontier of AI and its applications.

We are looking for creative thinkers with strong algorithm-design and programming skills, with expertise in, or passion for, environmental sustainability. We are seeking an engineer who excels at building robust, scalable software systems for machine learning research and applications. We particularly value strong technical experience with modern machine learning architectures, scaling models, and optimizing throughput.

Key responsibilities

• Design, implement, scale and evaluate state-of-the-art deep learning models (e.g., Transformers, GNNs) and software prototypes for sustainability-related problems.
• Build robust and scalable data processing and training pipelines to enable rapid research iterations.
• Report and present findings and developments clearly and efficiently, both internally and externally.
• Contribute to team collaborations to meet ambitious research and product goals.
• Engage with application and product needs, to inform research and engineering decisions.

About you

In order to set you up for success as a Research Engineer at Google DeepMind, we look for the following skills and experience:

• BSc, MSc or PhD/DPhil degree in computer science, mathematics, applied stats, machine learning or equivalent practical experience.
• Strong background in deep learning, with proven experience with relevant architectures (e.g., Transformers, GNNs, etc).
• Strong background in machine learning foundations (esp. supervised learning, probabilistic modeling, graph-based learning, and/or optimization).
• Excellent software engineering skills with a proven ability to build robust and scalable systems.
• Proficiency in deep learning frameworks like JAX, TensorFlow, or PyTorch is essential.
• Experience with either large-scale data processing frameworks (e.g., Apache Beam, Spark) or distributed training infrastructure.

In addition, the following would be an advantage:

• Core contributor to OSS projects.
• Experience with Python and its ecosystem.
• Experience with remote sensing data.
• Experience translating research innovations into product applications.
• Commitment/experience/expertise in weather/environmental sustainability and/or allied areas (e.g., fluid dynamics, complex systems, combinatorial optimization, remote sensing).
• A proven track record of publications in top-tier conferences and/or journals.

The US base salary range for this full-time position is between $182,000 - $215,000 + bonus + equity + benefits. Your recruiter can share more about the specific salary range for your targeted location during the hiring process.

Note: In the event your application is successful and an offer of employment is made to you, any offer of employment will be conditional on the results of a background check, performed by a third party acting on our behalf. For more information on how we handle your data, please see our Applicant and Candidate Privacy Policy.