Thinking Machines Lab

Research, Pre-Training Science

San Francisco

H1B sponsorship available$350k-$475kDetected 78 days ago

PythonExpressAlgorithmsMachine LearningDeep LearningTensorFlowPyTorchStatisticsA/B TestingLogisticsResearchCommunication

About the role

The role of pre-training researchers sits at the core of our roadmap.
This work advances the science of how large models learn from data.
You'll explore new pre-training methods, architectures, and learning objectives that make model training efficient, robust, and aligned with human goals.

Research and develop new methodologies for pre-training.
Collaborate with infrastructure and data teams to conduct large-scale experiments efficiently and reproducibly.
Ability to design, run, and analyze experiments thoughtfully, with demonstrated research judgment and empirical rigor.

Experience with distributed or high-performance computing environments.
Clarity in communication, an ability to explain complex technical concepts in writing.

we encourage you to apply even if you don't meet all preferred qualifications, but at least some:
A strong grasp of probability, statistics, and ML fundamentals.
You can look at experimental data and distinguish between real effects, noise, and bugs.
Prior experience training or analyzing large-scale models, or contributing to pre-training or foundation model research.
Experience designing or maintaining evaluation frameworks for large models.
Contributions to open datasets, research publications, or data tooling.
or, equivalent industry research experience.

Thinking Machines Lab's mission is to empower humanity through advancing collaborative general intelligence.

Depending on background, skills and experience, the expected annual salary range for this position is $350,000 - $475,000 USD.

Thinking Machines offers generous health, dental, and vision benefits, unlimited PTO, paid parental leave, and relocation support as needed.
Design data curricula and sampling strategies that improve learning dynamics and model generalization.
Proficiency in Python and familiarity with at least one deep learning framework (e.g., PyTorch, TensorFlow, or JAX).
Bachelor's degree or equivalent experience in Computer Science, Machine Learning, Physics, Mathematics, or a related discipline with strong theoretical and empirical grounding.

While we can't guarantee success for every candidate or role, if you're the right fit, we're committed to working through the visa process together.
We sponsor visas.

This listing is sourced directly from Thinking Machines Lab's careers page and normalized into a canonical job model.