Latest AI JobsBrowse Jobs
Thinking Machines logo

Research, Pre-Training Data

Thinking MachinesSan FranciscoResearch
Apply on company site
Salary
$350K - $475K USD
Job type
Full-time
Work mode
Remote
Posted
4 Aug 2026
Status
Open · checked 7d ago

About this role

Thinking Machines is hiring a Research, Pre-Training Data professional to build foundational AI that extends human will and judgment. This remote-friendly position in San Francisco suits individuals passionate about merging large-scale data engineering with machine learning research.

Role Overview

This position sits at the core of the organization's technological roadmap, blending fundamental scientific research with practical engineering to build next-generation AI models. Professionals in this group design and implement methods for sourcing, curating, and analyzing pre-training data to ensure optimal quality and performance. The work involves utilizing automated pipelines and human-in-the-loop processes while writing production-grade code.

Key Responsibilities

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 community forward by sharing code, datasets, and insights.

Requirements

Proficiency in Python and familiarity with at least one deep learning framework such as PyTorch, TensorFlow, or JAX, with comfort in debugging distributed training.
Bachelor’s degree or equivalent experience in Computer Science, Machine Learning, Physics, Mathematics, or a related discipline.
Clarity in communication with the ability to explain complex technical concepts in writing.

Nice to Have

A strong grasp of probability, statistics, and machine learning fundamentals.
Experience with curation, preprocessing, and analysis of large-scale text, code, or multimodal datasets.
Prior background in data engineering, dataset construction, or large-scale web data processing for machine learning models.
Knowledge of data ethics, safety, and licensing frameworks relevant to AI dataset creation.
Contributions to open datasets, research publications, or data tooling.
A PhD in Computer Science, Machine Learning, Physics, Mathematics, or a related field, or equivalent industry research experience.

Skills and Tools

Python
PyTorch
TensorFlow
JAX
Large-scale data processing systems
Distributed training debugging

Salary and Benefits

Annual salary range between $350,000 and $475,000 USD, depending on background, skills, and experience.
Generous health, dental, and vision benefits.
Unlimited paid time off (PTO).
Paid parental leave.
Relocation support as needed.
Visa sponsorship available for qualifying candidates.

About the Company

Thinking Machines builds artificial intelligence designed to extend human will and judgment. The organization trains frontier models using Inkling, develops Tinker to enable custom model creation, and crafts interfaces to broaden human-AI communication.

How to Apply

To express interest in this position, please submit an application directly on the official company careers page at thinkingmachines.ai.

Skills and tech stack

PythonPyTorchTensorFlow

Questions candidates ask

What is the salary range for the Research, Pre-Training Data role?

The expected annual salary range is $350,000 to $475,000 USD, depending on background, skills, and experience.

Where is this job located?

The role is based in San Francisco, California, with remote flexibility.

Does Thinking Machines offer visa sponsorship?

Yes, visa sponsorship is available for qualifying candidates who are the right fit for the role.

What technical skills are required?

Candidates must be proficient in Python and familiar with at least one deep learning framework like PyTorch, TensorFlow, or JAX, alongside experience with distributed training and scalable code.

What benefits are included?

Benefits include generous health, dental, and vision coverage, unlimited PTO, paid parental leave, relocation support, and visa sponsorship.

Ready to apply?

The application is on the official Thinking Machines careers page.

Apply on company site