Remote Jobs RockRemote Jobs Rock

Research, Mid Training

๐Ÿ•’ 6 days ago
PythonDeep Learning FrameworksData CurationSynthetic Data Generation

๐Ÿ“œ Description

  • Own the data by sourcing, curating, and synthesizing training-grade datasets.
  • Design and measure interventions to enhance model knowledge and retention.
  • Introduce new behaviors during mid-training and assess their impact on post-training.
  • Develop and maintain the quality pipeline for data filtering and scoring.
  • Iterate on mid-training recipes and evaluate their effects on performance metrics.
  • Debug training configurations and explore new methodologies for scaling.

๐Ÿ› ๏ธ Requirements

  • 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.
  • 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.

โœจ Benefits

  • Generous health benefits
  • Dental benefits
  • Vision benefits
  • Unlimited PTO
  • Paid parental leave
  • Relocation support as needed
Full job description

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.

Aisi

Research Engineer โ€“ Human Influence

๐Ÿ•’ 12 days ago
Aisi๐Ÿ‘ฅ 201 - 500 employees๐Ÿข Government Administration

The Research Engineer will focus on developing scalable solutions in LLM post-training and fine-tuning, particularly using Reinforcement Learning, to mitigate AI risks related to human influence.

Reinforcement LearningLLM Post-trainingMachine LearningPython
SuperAnnotate

Research Engineer

๐Ÿ“… Jul 20
SuperAnnotate๐Ÿ‘ฅ 201 - 500 employees๐Ÿข Computer Software

As a Research Engineer, you'll independently drive research initiatives, implementing methods and producing tangible outputs that enhance AI model performance and contribute to publications.

Machine LearningComputer ScienceResearch ExperiencePython
OpenAI

Research Engineer / Research Scientist, Health

๐Ÿ“… Aug 21
OpenAI๐Ÿ‘ฅ 10,000+ employees๐Ÿข Research Services

Seeking a hands-on researcher to develop and implement innovative health AI capabilities, driving impactful improvements in models and products.

Machine LearningAI ResearchPretrainingReinforcement Learning
OpenAI

Research Engineer / Research Scientist / AI Systems Engineer, RSI

๐Ÿ“… Aug 5
OpenAI๐Ÿ‘ฅ 10,000+ employees๐Ÿข Research Services

This role involves designing evaluations, improving model capabilities, and developing systems to automate research workflows at OpenAI.

LLM TrainingModel EvaluationsAgent SystemsSynthetic Data
OpenAI

Research Engineer / Research Scientist - Personal AGI, Personalization

๐Ÿ“… Aug 3
OpenAI๐Ÿ‘ฅ 10,000+ employees๐Ÿข Research Services

As a Research Engineer / Research Scientist on the Personalization-Memory team, you will research and develop improvements to memory usage and personalization in OpenAI's frontier models.

Memory UsagePersonalizationReinforcement LearningDataset Creation

Trusted by Remote Workers