# Research Engineer, Climate Modeling

**Company**: Ai2
**Location**: Seattle, WA
**Work arrangement**: hybrid
**Experience**: senior
**Job type**: full-time
**Salary**: $146,880 - $220,320
**Category**: Research
**Industry**: Technology

**Apply**: https://job-boards.greenhouse.io/thealleninstitute/jobs/8234447?utm_source=yubhub.co&utm_medium=jobs_feed&utm_campaign=apply
**Canonical**: https://yubhub.co/jobs/job_542544fe-cca

## Description

We seek a Research Engineer for a new three-year project to develop an AI-powered physics-informed hybrid climate model trained on historical data that is capable of multidecadal forecasts that are more accurate than the current state of the art.

You will join a small team to develop the desired NeuralGCM+ model and foster its uptake by the climate modeling community. This includes coupling it to an ML-based ocean model and designing approaches to demonstrably improve its skill in unseen climates.

Responsibilities:

- Adapting the training pipeline and data needs of NeuralGCM to run efficiently on Google Cloud resources and on NVIDIA GPUs.

- Documenting this 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 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.

Requirements:

- 2-4 years of experience building and deploying a relevant ML application in a practical or academic research setting and be fluent in Python and collaborative coding practices.

- A 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.

Benefits:

- Team members and their families are covered by medical, dental, vision, and an employee assistance program.

- Team members are able to enroll in our health savings account plan, our healthcare reimbursement arrangement plan, and our health care and dependent care flexible spending account plans.

- Team members are able to enroll in our company’s 401k plan.

- Team members will receive $125 per month to assist with commuting or internet expenses and will also receive $200 per month for fitness and wellbeing expenses.

- Team members will also receive up to ten sick days per year, up to seven personal days per year, up to 20 vacation days per year and twelve paid holidays throughout the calendar year.

- Team members will be able to receive annual bonuses and can participate in the long-term incentive plan.

## Skills

### Required
- Python
- ML
- cloud computing
- JAX

### Nice to have
- Ph.D. in computer or computational science
- atmospheric science or a related geophysical science
- applied mathematics/statistics
- experience writing machine learning code in JAX

---

Source: [Apply at job-boards.greenhouse.io](https://job-boards.greenhouse.io/thealleninstitute/jobs/8234447?utm_source=yubhub.co&utm_medium=jobs_feed&utm_campaign=apply)