Software Engineer

What you'll do

• Optimize GPU kernels, deep-learning operations, and distributed training for physics-aware transformer models.
• Build scalable, reproducible cloud pipelines and automate sensor-data ingestion and feature engineering.
• Turn research prototypes into production software and operationalize forecast systems with weather and AI teams.

What they ask for

• Degree in Atmospheric Science, Computer Science, or a related engineering or physical sciences field.
• 5+ years writing high-performance Python, C++, or Rust for GPU workloads.
• Experience building cloud-native applications and infrastructure, and deep fluency in at least one modern deep-learning framework.
• Track record shipping distributed training or inference systems at scale (100+ GPUs), plus proficiency with kernel profiling, memory bottlenecks, and modern DevOps.

Nice to have

• Experience with numerical weather prediction, remote-sensing data, or geospatial intelligence; background in physics-informed ML, simulation modeling, or data assimilation.
• Contributions to open-source ML or scientific computing frameworks.

Tools

CUDA, JAX, PyTorch, Ray, Kubernetes, Slurm, GCP, AWS

AI focus

Operationalize AI-native weather forecasting, including physics-aware transformer models, distributed training, and production forecast systems.