Research, Mid Training

Bison; San Francisco (CA); Presencial; USD 350.000 - 475.000; Jornada completa

Ventajas ofrecidas por este puesto de trabajo

Visa sponsorship

Health, dental, vision benefits

Unlimited PTO

Paid parental leave

Relocation support

Descripción de la vacante

Thinking Machines is hiring for a mid-training lead role in San Francisco to shape model capabilities by sourcing and curating data, building robust training pipelines, and measuring improvements across knowledge and behavior. This role blends research and practical engineering, working between pre-training and post-training to drive scalable data-centric AI.

You’ll collaborate with the post-training team to align behaviors with RL, develop data-quality systems, and explore data-driven

Formación

• Proficiency in Python and familiarity with deep learning frameworks (e.g., PyTorch, TensorFlow, or JAX).
• Bachelor’s degree or equivalent experience in Computer Science, Machine Learning, Physics, Mathematics, or related discipline.
• Clarity in communication, ability to explain complex technical concepts in writing.

Responsabilidades

• Own the data: source, curate, and synthesize data; build pipelines to filter, deduplicate, verify, and rewrite material into training-grade datasets.
• Improve what the model knows: design interventions with targeted corpora and QA over source documents.
• Instill behaviors and set the prior: evaluate mid-training vs post-training impacts and coordinate with RL teams.
• Build the quality pipeline: develop automatic filtering and scoring; calibrate classifiers and LLM-based judges.
• Develop and tune the recipe: iterate on datasets, training stages, annealing schedules, and hyperparameters.
• Iterate on evals: define evaluations, optimize, and ensure they reflect meaningful results.
• Debug and understand: diagnose results and deepen understanding for future problems.
• Scale and explore: measure performance vs data size and explore new data approaches.

Conocimientos

Python proficiency

Deep learning frameworks

Communication clarity

Educación

Bachelor’s degree or equivalent in CS/ML/Physics/Math

Herramientas

PyTorch

TensorFlow

JAX

Descripción del empleo

About Thinking Machines

The mission of Thinking Machines is to build AI that extends human will and judgment. We are training frontier models with Inkling, developing Tinker to let people make models their own, and crafting interfaces that broaden human-AI communication. We believe the future worth building is human, and we're hiring people who want to build it.

About the Role

Mid-training is a step between pre-training and post-training, where we take a base model and train it into the foundation for reasoning. This role owns the late-stage training responsibility that shape what our models are fundamentally capable of, including things like synthetic data strategies, the data mix, quality uplift, context extension and capabilities across coding, math, reasoning, and so on.

This role blends fundamental research and practical engineering, as we do not distinguish between the two internally. It's an excellent fit for someone comfortable working across the boundary of pre-training and post-training, and who wants to shape what our models can do at their core.

What You’ll Do

• Own the data. Decide what the model needs to see for each capability and each area of knowledge, then source, curate, and synthesize it. Build the pipelines that filter, deduplicate, verify, and rewrite raw material into training-grade datasets. You'll be responsible for producing data that actually moves the model, and for making sure it reflects how people really use these models, not just clean benchmark-style tasks.

• Improve what the model knows. Design and measure interventions that increase knowledge: targeted corpora, synthetic rephrasings and QA over source documents, knowledge-dense mixes. Characterize how knowledge scales with data and when it is retained through post-training versus forgotten.

• Instill behaviors and set the prior. Introduce new behaviors during mid-training and measure them the right way: not just whether an eval goes up, but whether post-training becomes easier, more sample-efficient, and more stable as a result. Work closely with the post-training team to decide which behaviors belong in mid-training and which belong in RL.

• Build the quality pipeline. Own the automatic filtering and scoring stack: train and calibrate quality classifiers and LLM-based judges, build verifiers for synthetic data, and use them to raise the quality bar of the mix by a lot, not a little. Give the team fine-grained control over data attributes (difficulty, domain, format, style, correctness) and measure the effect of each.

• Develop and tune the recipe. Iterate on mid-training recipes: the collection of datasets, training stages, annealing schedules, and hyperparameters. Measure how recipe choices affect metrics, including downstream of post-training.

• Iterate on evals. Mid-training involves a never-ending loop of defining a set of evaluations, optimizing them, and then realizing your existing evals don't capture what matters. You'll be responsible for both making numbers go up and making sure the numbers are meaningful.

• Debug and understand. While tuning the details of a training configuration, we often observe results that don't quite make sense. You'll be responsible for both getting things to work and developing a deeper understanding we can bring to the next problem.

• Scale and explore. Mid-training will involve a combination of scaling existing methodologies and developing new ones. We'll want to both measure how performance scales with dataset size and explore using completely different kinds of training data.

Skills and Qualifications

Required qualifications:

• Proficiency in Python and familiarity with deep learning frameworks (e.g., PyTorch, TensorFlow, or JAX). Comfort debugging distributed training and writing code that scales.

• Bachelor’s degree or equivalent experience in Computer Science, Machine Learning, Physics, Mathematics, or a related discipline with strong theoretical and empirical grounding.

• Clarity in communication, an ability to explain complex technical concepts in writing.

Preferred qualifications

(we encourage you to apply if you meet some but not all of these):

• A strong grasp of probability, statistics, and ML fundamentals. You can look at experimental data and distinguish between real effects, noise, and bugs.

• Experience building or owning training datasets for large models, including both synthetic data pipelines and real user data (collection, curation, filtering, or mixture design).

• Experience with knowledge injection, continued pre-training, or domain adaptation of large models, and with measuring what a model knows.

• Experience building model-based data quality systems: quality classifiers, LLM judges, or verification and rewriting pipelines operating at the scale of billions of tokens.

• Experience generating synthetic data and understanding its failure modes (diversity collapse, contamination, teacher artifacts).

• Research or engineering contributions in alignment, data-centric AI, or human-AI collaboration.

Logistics

• This is an IC role within Thinking Machines’ Research Team. You will report directly to our Head of Research, and work most closely with our small group of post-training researchers.

• Location: This role is based in San Francisco, California.

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

• Visa sponsorship: We sponsor visas. 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.

• Benefits: Thinking Machines offers generous health, dental, and vision benefits, unlimited PTO, paid parental leave, and relocation support as needed.

As set forth in Thinking Machines' Equal Employment Opportunity policy, we do not discriminate on the basis of any protected group status under any applicable law.