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 weve 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 torevolutionize 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. Notablyour 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 humanitys most useful inventions. At Google DeepMind were 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 $182000 - $215000 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 ourApplicant and Candidate Privacy Policy.