Research Engineer, Climate Modeling

The role

You will develop an AI-powered physics-informed hybrid climate model to improve multidecadal forecast accuracy. This involves adapting training pipelines for cloud and GPU resources, coupling models, and developing open-source code.

What you will do

• Adapting the training pipeline and data needs of NeuralGCM to run on Google Cloud resources and NVIDIA GPUs
• Documenting the training pipeline and distributing code on a GitHub repo
• Implementing and testing a strategy for coupling NeuralGCM with the chosen ML ocean model
• Open-source code development including daily meetings, code review, and design documents
• Develop and present tutorials on NeuralGCM+ to interested user groups

Required qualifications

• 2-4 years of experience building and deploying a relevant ML application in a practical or academic research setting
• Fluent in Python and collaborative coding practices
• B.Sc. in computer or computational science, atmospheric science or a related geophysical science, or applied mathematics/statistics
• Experience in developing ML training code for use in a cloud computing environment
• Demonstrated excellent communications skills
• Must be able to remain in a stationary position for long periods of time
• Ability to communicate information and ideas so others will understand
• Ability to observe details at close range
• Can work under deadlines

Nice to have

• Ph. D. in computer or computational science, atmospheric science or a related geophysical science, or applied mathematics/statistics, with at least one peer-reviewed first-authored publication or acce
• Formal training and/or practical experience with physically-based atmospheric, oceanic, or related model development
• Experience writing machine learning code in JAX

Benefits

• Medical, dental, vision, and an employee assistance program
• Health savings account plan
• Healthcare reimbursement arrangement plan
• Health care and dependent care flexible spending account plans
• 401k plan
• $125 per month to assist with commuting or internet expenses
• $200 per month for fitness and wellbeing expenses
• Up to ten sick days per year
• Up to seven personal days per year
• Up to 20 vacation days per year
• Twelve paid holidays throughout the calendar year
• Annual bonuses and participation in the long-term incentive plan

Before you apply

• Applications will begin being reviewed on October 28, 2026
• Employer participates in E-Verify to confirm authorization to work in the U.S.

About the company

Ai2 is a Seattle based non-profit AI research institute founded in 2014 by the late Paul Allen. Its mission is building breakthrough AI to solve the world’s biggest problems.