Research Engineer, Climate Modeling
Core
Develop an AI-powered physics-informed hybrid climate model (NeuralGCM+) capable of multidecadal forecasts more accurate than current state-of-the-art models.
Role type
Research Engineer (Climate Modeling & AI)
Builds
Open-source 'AI-first' climate model, training pipelines, and coupled ML ocean models.
Domain
Climate Science / Machine Learning / Atmospheric Physics
Deliverable
production ML models
Required skills
Python, ML training code, cloud computing, JAX, collaborative coding, documentation, tutorial development
Preferred skills
PhD in computational/atmospheric science, peer-reviewed ML publications, physically-based model development, JAX expertise
Technologies
JAX, Google Cloud, NVIDIA GPUs, TPUs, GitHub
Responsibilities
Adapt NeuralGCM training pipeline for cloud/GPU efficiency, implement coupling with ML ocean model, develop and present tutorials, document code and distribute on GitHub
Seniority
Mid-level (2-4 years experience)
