Member of Technical Staff — ML Research, Multimodal

Causal Labs

San Francisco Full-time 3 mo ago
Workplace
On-site
Location
San Francisco
Who can apply
Based in the United States: you usually need the right to work there

Check your CV against this job

Free · No signup · A 0–100 ATS match score and the keywords you're missing.

Check my CV free

$14.99/month in most countries, cancel anytime. Already have an account? Log in

About the role

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 researchers who are excited to tackle unsolved problems. Predicting how physical systems evolve means learning from observations that language and vision models were not built for — sparse sensors, point clouds, hyperspectral imagery, physical fields — at a scale that dwarfs what is used to train even today's frontier LLMs. Your mission is to design the architectures and training recipes that turn these multimodal observations into a model that predicts the future of the physical world.

Responsibilities

* Design and implement novel model architectures and training algorithms for learning from massive, multimodal physical data

* Solve core modeling problems unique to physical prediction: encoding heterogeneous and irregularly-sampled modalities, stable long-horizon rollouts, and probabilistic forecasting

* Run experiments and ablations that connect modeling and data decisions to predictive skill, including which data sources and mixtures most improve the model

* Work across the full ML stack — data, model, eval, and infrastructure — to take ideas from prototype to scaled training runs

* 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, with depth in at least one relevant domain (e.g. sequence or world models, computer vision, sensor fusion, generative modeling, physics-informed NNs)

* Experience training large-scale models and the ability to understand experimental results through careful analysis and ablation studies

* Familiarity with distributed training and the systems considerations of scaling models

* A track record of turning open-ended research problems into production models

About jobs in the United States

Open jobs
191,766 open jobs in the United States, 31,009 of them posted in the last 7 days. See all Jobs in the United States
Typical advertised pay
$85,000 – $185,000 a year for the middle half of the 69,926 job ads from 7,711 employers that state a salary in US dollars (median $134,000). This is base pay as advertised, not total compensation.

Tools and guides for your application

Get matched & apply with FindAJobAI

Upload your resume once. We score every job against your profile, tailor your resume and cover letter, and autofill the application.