Senior Scientific Machine Learning Engineer – Earth-2
Core
Developing AI frameworks (NVIDIA PhysicsNeMo, Earth2Studio) for weather, climate, and earth system modeling to democratize professional-grade climate AI.
Role type
Senior Scientific Machine Learning Engineer
Builds
Open models, libraries, and frameworks for scientific ML technology and workflows
Domain
Climate science, weather modeling, deep learning
Deliverable
production ML models
Required skills
Python programming, deep learning frameworks (PyTorch, Tensorflow, JAX), DNN architectures (graph networks, diffusion models, reinforcement learning), scientific visualization, modular software design, numeric libraries, containers
Preferred skills
HPC programming (OpenMPI, NCCL), CUDA/GPU kernel programming, multi-node systems, nonlinear simulation tools, xarray, zarr, regridding
Technologies
PyTorch, Tensorflow, JAX, OpenMPI, NCCL, CUDA, xarray, zarr
Responsibilities
Develop leading machine learning frameworks for academic and industrial partners, validate applications built using the framework, integrate new functionalities into the platform, implement and experiment with new deep learning ideas for weather & climate AI
Seniority
Senior, hands-on IC