Machine Learning - Research

Location
San Francisco
Workplace
On-site
Compensation
$200k – $200k + equity
Visa
Visa Sponsorship Available

About this role

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.

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.

What happens next

Skip the application pile. I get you in front of the people who decide.

Confirm the fit

A few questions to make sure this role is the right shape for you. Two minutes.

I pitch you to the company

I write the intro, send it to the founder, and handle the back-and-forth.

A meeting lands on your calendar

When the company wants to meet, I get the call on your calendar. You just show up.

Culture & values

The team is composed of researchers and engineers from self-driving, drug discovery, and robotics.

Team members have experience from Google DeepMind, Cruise, Waymo, Meta, Nabla Bio, and Apple.

They believe general causal intelligence will be the most important technical breakthrough for civilization.

The culture is mission-driven and ambitiously oriented toward civilization-scale breakthroughs.

The organization emphasizes interdisciplinary collaboration across diverse domain backgrounds and industry experience.

Know someone who'd be great for this?