Member of Technical Staff
Causal Labs
Member of Technical Staff
Causal Labs
Trains physics AI models to forecast weather
Our mission is general causal intelligence; AI that is capable of (1) predicting the future and (2) identifying the actions to alter it.
To achieve this breakthrough, we are building a Large Physics foundation Model (LPM) because physical systems, unlike text or images, are governed by verifiable cause and effect. We believe that scaling on physics will enable an understanding of causality required to predict and control physical systems, starting with weather.
Our founding team has built and deployed AI against the physical world in robotics, drug discovery, and particle physics at institutions like DeepMind, Waymo, Cruise, Insitro, Nabla Bio, and CERN.
We look for infrastructure engineers who are excited to tackle unsolved problems. Everything we do — training, evaluation, serving — runs on our GPU fleet. Your mission is to design, build, and operate the supercomputing environment underneath it all, delivering performant, reliable, and cost-efficient compute to ensure research is able to iterate rapidly at scale.
Responsibilities
Design, deploy, and operate large distributed GPU clusters end to end: provisioning, imaging, upgrades, and capacity planning
Extend scheduling and orchestration systems (e.g. Kubernetes, Slurm) for topology-aware placement, preemption, quotas, and multi-tenancy across training and inference workloads
Build software that abstracts cluster management and presents a unified, self-serve interface to researchers and engineers
Own cluster storage and artifact paths for checkpoints and logs, with clear retention and lineage
Monitor and continuously improve reliability and error recovery; build the observability to catch failures before researchers do
Partner with researchers to unblock large-scale runs and advise on performance and placement trade-offs
What we're looking for
We value a relentless approach to problem-solving, rapid execution, and the ability to quickly learn in unfamiliar domains.
Experience operating large-scale GPU clusters and container orchestration frameworks (e.g. Kubernetes, Slurm, Docker)
Strong systems background: Linux, networking, storage, infrastructure-as-code
Knowledge of cloud platforms (GCP, AWS, or Azure) and their ML/AI service offerings
Understanding of monitoring, logging, observability, and version control best practices for ML systems
Familiarity with CUDA/NCCL and performance profiling for distributed workloads
Owns deliverables end-to-end, from requirements through autonomous execution
Trains physics AI models to forecast weather
Causal Labs is an AI research company building a foundation model of physical systems for industries whose operations turn on the weather. The company trains on atmospheric sensor and observation data rather than text or images, a domain where cause and effect are governed by physics and yesterday's prediction can be scored against what happened. It runs early pilots with commercial partners and pairs researchers with engineers who deploy the models on customer data. Unlike conventional forecasters, which report what conditions are likely, it is also trying to identify which interventions would change an outcome. The company's goal is to make AI reason about cause and effect by grounding it in the most heavily observed physical system available.
Explore more opportunities at Causal Labs