Senior Scientific Machine Learning Engineer – Earth-2

Nvidia

Remote Actively hiring Verified listing
Remote, US · Santa Clara, CA Posted 10 days ago $152,000–$241,500 / year
View original post Log in to save

At a glance

AI generated

TL;DR

Join NVIDIA’s deep learning and HPC platforms team as a senior software engineer to contribute to the development of AI frameworks for the NVIDIA Earth-2 platform. You will collaborate with leading experts to build and enhance machine learning frameworks like PhysicsNeMo and Earth2Studio, focusing on weather and climate modeling. Your responsibilities include validating applications built using these frameworks on NVIDIA’s products, integrating new functionalities from internal or external projects, and staying updated with cutting-edge deep learning techniques to implement innovative solutions. Ideal candidates have a BS/MS (PhD preferred) in computer science or related fields, 5+ years of experience, strong Python skills, familiarity with deep learning frameworks like PyTorch and TensorFlow, and expertise in applying machine learning to real-world weather and climate scenarios.

Skills

Python PyTorch Tensorflow JAX OpenMPI NCCL CUDA GPU HPC Deep Learning DNN Graph Networks Diffusion Models Reinforcement Learning xarray zarr Weather and Climate Data Stores

What you'll do

  • Develop leading machine learning frameworks for NVIDIA PhysicsNeMo and NVIDIA Earth2Studio.
  • Validate applications built using the framework on NVIDIA’s products and integrate new functionalities.
  • Stay updated with latest deep learning techniques to enhance NVIDIA's Earth-2 technologies.
  • Implement and experiment with new ideas to develop and enhance AI in weather & climate modeling.
  • Use Python programming skills to create scientific ML technology and workflows.

What we're looking for

  • BS or MS degree in computer science, mathematics, or related field; PhD preferred.
  • 5+ years of experience developing machine learning frameworks and technologies.
  • Strong Python programming skills with expertise in deep learning frameworks like PyTorch.
  • Deep knowledge of DNN architectures and ML techniques, including graph networks and reinforcement learning.
  • Experience applying machine learning to real-world weather and climate scenarios.
  • Familiarity with scientific visualization tools and multi-node systems for data parallelism.

Market check

Salary context

This $152,000–$241,500 range sits above 45% of similar postings on FindRole.

Peer median band

$159,500–$236,450

Median floor and ceiling across peers.

Typical midpoint (25–75%)

$164,613–$241,675

Middle half of comparable postings.

Based on 240 comparable postings.

* 240 is the maximum number of comparable postings sampled.

Employer

About Nvidia

Nvidia is a leading designer of graphics processing units (GPUs) and system-on-chip units, powering gaming, professional visualization, data centers, and artificial intelligence workloads. Industry: Semiconductors & AI Computing

Nvidia currently has 801 open roles on FindRole.

Listed pay typically runs $184,000–$287,500 across 797 roles with salary data.

Most-posted roles

View all roles at Nvidia

More like this

Similar roles

Senior Machine Learning Engineer

Adobe

San Francisco, US 43 days ago $211,800–$306,625
Python PyTorch TensorFlow Machine_Learning Deep_Learning Data_Science CI/CD Mentorship Collaboration Research_and_Development Product_Integration Cross_Functional_Teams

Senior Machine Learning Engineer

Adobe

San Jose, US 52 days ago $183,300–$265,350
Python PyTorch TensorFlow Docker AWS Azure MLOps CI/CD PostgreSQL Adobe Experience Platform Marketo Engage Journey Optimizer LLMs RAG semantic embeddings agentic AI workflows ML inference systems Prometheus Grafana

Senior Machine Learning Engineer

Adobe

San Francisco, US 43 days ago $211,800–$306,625
Python PyTorch TensorFlow Machine_Learning Deep_Learning CI/CD MLOps Data_Science Collaboration_Tools Version_Control_Systems Cloud_Platforms API_Development Database_Management Model_Evaluation Technical_Guidance

Senior Machine Learning Engineer

Adobe

San Jose, US 11 days ago $183,300–$265,350
Python Java PyTorch TensorFlow JAX AWS Azure Postgres Redis Elasticsearch Snowflake CI/CD LLMs APIs Sandboxed Execution Memory Systems Observability Security ML-Ops Kubernetes Docker

Senior Machine Learning Engineer

Zillow

Remote (Remote-Usa, US) 36 days ago $171,700–$274,300
Python AWS Spark Databricks Airflow LangGraph LangChain AgentsSDK CI/CD Prometheus Grafana PostgreSQL Kubernetes
Remote