AI-Driven Climate Modeling Research Engineer

The Allen Institute for Artificial Intelligence; Seattle (WA); On-site; USD 147,000 - 220,000; Full time

Job summary

The Allen Institute for Artificial Intelligence (Ai2) in Seattle is seeking a Research Engineer for a 3-year project to develop NeuralGCM+—an open-source AI-first climate model. You will adapt training pipelines to run on Google Cloud and NVIDIA GPUs, document pipelines, couple models with ML ocean components, and contribute to open-source code with a collaborative team.

We value 2–4 years of ML experience in academia or industry, a B.Sc.

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.
• Experience developing ML training code for use in a cloud computing environment.
• Demonstrated excellent communications skills.

Responsibilities

• Adapt the training pipeline and data needs of NeuralGCM to run efficiently on Google Cloud resources and on NVIDIA GPUs.
• Document this training pipeline and distribute code on a GitHub repo.
• Implement and test a strategy for coupling NeuralGCM with the chosen ML ocean model.
• Open-source code development as part of a tightly connected team, including daily meetings, code review, design documents and external collaborators.
• Develop and present tutorials on NeuralGCM+ to interested user groups.

Skills

Python

Cloud computing

Communication skills

Education

B.Sc. in related field

Job description

The Allen Institute for Artificial Intelligence (Ai2) in Seattle is seeking a Research Engineer for a 3-year project to develop NeuralGCM+—an open-source AI-first climate model. You will adapt training pipelines to run on Google Cloud and NVIDIA GPUs, document pipelines, couple models with ML ocean components, and contribute to open-source code with a collaborative team.

We value 2–4 years of ML experience in academia or industry, a B.Sc.