Machine Learning - Research
Causal Labs
Location
San Francisco
Employment Type
Full time
Location Type
On-site
Department
Engineering
Our mission is general causal intelligence, AI that is capable of (1) predicting the future and (2) identifying the optimal actions to change that future.
To achieve this breakthrough, we are building a Large Physics foundation Model (LPM) because domains governed by physics have inherent cause and effect relationships, unlike visual or textual data.
Weather is the ideal training ground for an LPM. It is the most well-observed physical system, offering rapid, objective ground truth feedback from sensory observations and data at a scale that dwarfs what is used to train today’s LLMs.
Causal Labs is a team of researchers and engineers from self-driving, drug discovery, and robotics - including Google DeepMind, Cruise, Waymo, Meta, Nabla Bio, and Apple - who believe general causal intelligence will be the most important technical breakthrough for civilization.
We look for researchers who are excited to tackle unsolved problems.
Our research challenges offer an opportunity to build powerful models grounded in observable feedback and verifiable ground truth. If you have experience doing frontier research and training large-scale models from scratch in related fields such as language and vision models, robotics, biology – join us.
Responsibilities
Work across the full ML stack (data, model, eval, and infrastructure)
Implement novel model architectures and training algorithms
Build data pipelines and training infrastructure for massive, petabyte-scale, multimodal datasets
Rapidly iterate on experiments and ablations
Stay up-to-date on research to bring new ideas to work
What we’re looking for
We value a relentless approach to problem-solving, rapid execution, and the ability to quickly learn in unfamiliar domains.
Strong grasp of machine learning fundamentals, and depth in at least one core domain (e.g. Computer Vision, Sensor Fusion, Language Models, Physics-informed NNs)
Experience training models and an ability to understand experiment results through careful analysis and ablation studies.
Experienced at writing and optimizing massive petabyte-scale data pipelines.
Familiarity with distributed training and inference.
[bonus] Familiarity with meteorology, computational fluid dynamics, and/or numerical simulations.
You don’t have to meet every single requirement above.