Senior Scientific Machine Learning Engineer – Earth-2

About the role

NVIDIA is seeking individuals passionate about building AI frameworks for weather and climate modeling. The successful candidate will work on developing and enhancing NVIDIA’s Earth-2 technologies, focusing on machine learning frameworks like NVIDIA PhysicsNeMo and NVIDIA Earth2Studio.

Responsibilities

• Work with internal project teams to validate applications built using the framework on NVIDIA’s products.
• Integrate new functionalities from internal or external projects into the platform.
• Stay up to date with the latest research and innovations in deep learning techniques.
• Implement and experiment with new ideas to develop and enhance NVIDIA's Earth-2 technologies, with a focus on weather & climate AI.

Requirements

• BS or MS degree (PhD preferred) in computer science, mathematics, computational science/engineering, or related technical field or equivalent experience.
• 5+ years of relevant experience.
• Strong Python programming skills.
• Familiarity with containers, numeric libraries, modular software design.
• Deep knowledge of state-of-the-art DNN architectures and machine learning techniques and algorithms (graph networks, diffusion models, reinforcement learning etc.).
• Experience with developing or using major deep learning frameworks (PyTorch, Tensorflow, JAX etc.).
• Experience with machine learning techniques to solve real-world scenarios in weather/climate.
• Experience with scientific visualization.
• Strong analytical skills with a bias for action.
• Good time-management and organization skills to thrive in a fast-paced, dynamic environment.
• Solid written and oral communications skills.
• Good teamwork and interpersonal skills.

Qualifications

• Experience using multi-node systems with data-parallel and model-parallel programming, performance optimization.
• Experience with HPC programming models (OpenMPI, NCCL), and/or CUDA or GPU kernel programming.
• Published papers in the field of AI in scientific computing, especially in weather & climate applications.
• Familiarity with common tooling in the Earth-2 ecosystem (xarray, zarr, regridding, weather & climate data stores, etc.).

Skills

Experience using multi-node systems with data-parallel and model-parallel programming, performance optimization. Experience with HPC programming models (OpenMPI, NCCL), and/or CUDA or GPU kernel programming. Published papers in the field of AI in scientific computing, especially in weather & climate applications. Familiarity with common tooling in the Earth-2 ecosystem (xarray, zarr, regridding, weather & climate data stores, etc.).

Benefits

Base salary range: $152,000 - $241,500 for Level 3, and $184,000 - $287,500 for Level 4. Eligible for equity and benefits.

Pay

Base salary will be determined based on your location, experience, and the pay of employees in similar positions.

Schedule

Full-time position.