Research, Pre-Training Data

Thinking Machines Lab; San Francisco (CA); On-site; USD 350,000 - 475,000; Full time

Benefits offered by this job

Health, dental, and vision benefits

Unlimited PTO

Paid parental leave

Relocation support

Job summary

A leading AI research organization in San Francisco seeks a pre-training researcher to blend research with data engineering. The ideal candidate will design and implement methods for sourcing and curating data, develop metrics for data quality, and collaborate across teams to enhance AI model performance. Candidates should have a bachelor's degree in a relevant field and proficiency in Python, with familiarity in deep learning frameworks being a plus. Competitive compensation and comprehensive benefits are offered.

Qualifications

• Experience with debugging distributed training.
• Strong theoretical and empirical grounding in ML.
• Ability to explain complex technical concepts.

Responsibilities

• Design and implement techniques for curating large-scale data.
• Develop metrics to measure data quality.
• Collaborate with teams to scale data processing systems.

Skills

Proficiency in Python

Familiarity with deep learning frameworks like PyTorch

Clarity in communication

Education

Bachelor’s degree in Computer Science or related disciplines

Job description

Thinking Machines Lab's mission is to empower humanity through advancing collaborative general intelligence. We're building a future where everyone has access to the knowledge and tools to make AI work for their unique needs and goals.

We are scientists, engineers, and builders who’ve created some of the most widely used AI products, including ChatGPT and Character.ai, open-weights models like Mistral, as well as popular open source projects like PyTorch, OpenAI Gym, Fairseq, and Segment Anything.

About the Role

The role of pre-training researchers sits at the core of our roadmap. This work blends research with large-scale data engineering to help assemble the pre-training datasets and data systems that underpin the next generation of AI models. You’ll design and implement methods for sourcing, curating, and analyzing pre-training data for quality and performance.

You’ll work with automated pipelines and human-in-the-loop processes, contributing both scientific insight and production-grade code. It’s ideal for someone who enjoys working at the intersection of data, machine learning, and systems, and who’s excited by the challenge of shaping frontier AI.

This role blends fundamental research and practical engineering, as we do not distinguish between the two roles internally. You will be expected to write high-performance code and read technical reports. It’s an excellent fit for someone who enjoys both deep theoretical exploration and hands-on experimentation, and who wants to shape the foundations of how AI learns.

Note: This is an "evergreen role" that we keep open on an on-going basis to express interest in this research area. We receive many applications, and there may not always be an immediate role that aligns perfectly with your experience and skills. Still, we encourage you to apply. We continuously review applications and reach out to applicants as new opportunities open. You are welcome to reapply if you get more experience, but please avoid applying more than once every 6 months. You may also find that we put up postings for singular roles for separate, project or team specific needs. In those cases, you're welcome to apply directly in addition to an evergreen role.

What You’ll Do

• Design and implement techniques for curating, sourcing, and filtering large-scale text, code, and multimodal data.
• Develop data quality metrics and analysis to measure coverage, diversity, and representativeness across sources.
• Collaborate with research and infrastructure teams to scale data processing systems efficiently and reproducibly.
• Investigate and mitigate data risks, including privacy, safety, and licensing concerns, to ensure responsible and ethical data use.
• Continuously evaluate dataset improvements by analyzing their downstream effects on model learning and behavior.
• Publish and present research that moves the entire community forward. Share code, datasets, and insights that accelerate progress across industry and academia.

Skills and Qualifications

• Proficiency in Python and familiarity with at least one deep learning framework (e.g., PyTorch, TensorFlow, or JAX). Comfortable with 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 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.
• Experience with curation, preprocessing, and analysis of large-scale text, code, or multimodal datasets.
• Prior experience in data engineering, dataset construction, or large-scale web data processing for machine learning models.
• Experience evaluating or improving training data quality and knowledge of data ethics, safety, and licensing frameworks relevant to AI dataset creation.
• Contributions to open datasets, research publications, or data tooling.
• PhD in Computer Science, Machine Learning, Physics, Mathematics, or a related discipline with strong theoretical and empirical grounding; or, equivalent industry research experience.

Logistics

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

Create a Job Alert

Interested in building your career at Thinking Machines Lab? Get future opportunities sent straight to your email.

Accepted file types: pdf, doc, docx, txt, rtf

Education

School * Select...

Degree * Select...

Select...

Start date year

End date year

LinkedIn Profile Link *

Please provide the URL to your LinkedIn; if you don't have one, please write "none".

Github Link *

Please provide the URL to your Github; if you don't have one, please write "none".

Personal Website About You *

Please provide the URL to your personal website, Google Scholar, etc if you have one. Put "none" if you do not.

Current Company *

Please tell us the name of your current employer (today if you are employed). Put "none" if this does not apply to you; for example, if you are in school or not currently employed -- this does not disqualify you. Feel free to enter previous roles in the field below in "Past Company 1".

Current Title or Role *

Please enter your current title at your current employer. If you are not currently employed (or in school etc) please enter "none" and feel free to enter previous roles in the field below in "Past Company".

Past Company 1 *

Please enter the Company name of your most recent previous employer. If you have not worked at another company before your current one, please enter “none”.

Past Company Title or Role *

Please enter your title at your most recent previous employer. If you have not worked at another company before your current one, please enter “none”.

Past Company 2

If you would like, please enter the Company name of your second previous employer. If you have not worked at another company before your current or previous one, please enter “none” or skip this question.

Past Company Title or Role 2

If you would like, please enter the Title or Job of your second previous employer. If you have not worked at another company before your current or previous one, please enter “none” or skip this question.

What domains of research do you have expertise in? *

Pre-training Data

Pre-training Science

Pre-training

Scaling Laws

Optimization

Audio

Perception

Vision

Robotics

Post-Training

Post-Training Data

Human Data

Synthetic Data

Reasoning

Safety

Alignment

Other

Select all that apply where you have actively completed work in and would be able to interview for in a technical interview. This will help us when picking between projects

If you selected other, what areas did we not include that you have expertise in? *

If you completed research under an advisor, such as through a PhD program or masters program where you published, who was your advisor?

First name and last name of your Advisor / what program this was

Do you have any links to publications we should read?

Links to any publications, please list here

(Optional) List 3 projects you're proud of.

Please list 3 projects you're proud of, using 1 sentence each. Feel free to add a link if helpful. We will send this to the hiring team when reviewing your application.

Will you now or in the future require sponsorship for employment visa status in the United States? * Select...