Thinking Machines Lab
Research, Pre-Training Data
San Francisco
Stay score
odds of building a lasting career here
Sponsors, but it's cap-subject — you still face the weighted lottery (~61% per draw at Level IV). Good if you win; have a cap-exempt backup on your list.
Lottery odds assume a STEM candidate.
Personalize to your clock →Employer immigration record
from this employer's Department of Labor filings
Files H-1B transfers
Sourced from Department of Labor LCA, PERM and prevailing-wage disclosure data. Employer matching is by name, so figures may be split across an employer's legal entities. Absence of a filing means none appears in our copy of the data, not that none exists.
Community outcomes
No reports yet — be the first to help the next applicant.
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 work with automated pipelines and human-in-the-loop processes, contributing both scientific insight and production-grade code.
Responsibilities
- Design and implement techniques for curating, sourcing, and filtering large-scale text, code, and multimodal data.
- Collaborate with research and infrastructure teams to scale data processing systems efficiently and reproducibly.
Requirements
- Clarity in communication, an ability to explain complex technical concepts in writing.
- 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.
- Experience evaluating or improving training data quality and knowledge of data ethics, safety, and licensing frameworks relevant to AI dataset creation.
Nice to have
- we encourage you to apply even if you don't meet all preferred qualifications, but at least some:
Compensation
- Depending on background, skills and experience, the expected annual salary range for this position is $350,000 - $475,000 USD.
Benefits
- Thinking Machines offers generous health, dental, and vision benefits, unlimited PTO, paid parental leave, and relocation support as needed.
- Develop data quality metrics and analysis to measure coverage, diversity, and representativeness across sources.
- Continuously evaluate dataset improvements by analyzing their downstream effects on model learning and behavior.
- 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.
Company info
- The mission of Thinking Machines is to build AI that extends human will and judgment.
Visa & Work Authorization
- 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.
This listing is sourced directly from Thinking Machines Lab's careers page and normalized into a canonical job model.