Research, Pre-Training Data

Thinking Machines Lab
San Francisco
Workplace: HybridFull timeUSD 350,000 - 475,000 annuallyFunction: Education & TrainingEducation: bachelorsSkills: ["Communication","Collaboration","Problem-solving","Analytical thinking"]

Build and evaluate pre-training datasets and data systems for next-generation AI models. Design methods to source, curate, filter, and measure quality across large-scale text, code, and multimodal data, using automated pipelines and human-in-the-loop processes. Collaborate with research and infrastructure teams to scale reproducible data processing, mitigate privacy/safety/licensing risks, and publish research while contributing production-grade code.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
Thinking Machines Lab
Thinking Machines Lab
20 hours ago

Research, Pre-Training Data

✓ Verified Job

Canonical indexed version, validated from employer's careers page.

Source: Company careers pageValidated by: Fursa AI
Last checked: 5 hours agoStatus: Live

Job Summary

Build and evaluate pre-training datasets and data systems for next-generation AI models. Design methods to source, curate, filter, and measure quality across large-scale text, code, and multimodal data, using automated pipelines and human-in-the-loop processes. Collaborate with research and infrastructure teams to scale reproducible data processing, mitigate privacy/safety/licensing risks, and publish research while contributing production-grade code.
Location: San Francisco
Workplace: Hybrid
Employment Type: Full time
Job Function: Education & Training

Key Responsibilities

  • •Design and implement techniques for curating, sourcing, and filtering large-scale text, code, and multimodal data.
  • •Develop data quality metrics 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, for responsible and ethical data use.
  • •Continuously evaluate dataset improvements by analyzing downstream effects on model learning and behavior, and publish research while sharing code and datasets.

Pay and Benefits

Salary: USD 350,000 - 475,000 annually
Perks:Health InsuranceDentalVisionPaid LeaveRelocation

Key Requirements

  • •Proficiency in Python and familiarity with at least one deep learning framework (PyTorch, TensorFlow, or JAX), with comfort debugging distributed training and writing scalable code.
  • •Bachelor’s degree or equivalent experience in Computer Science, Machine Learning, Physics, Mathematics, or a related discipline with strong theoretical and empirical grounding.
  • •Ability to clearly communicate complex technical concepts in writing.
  • •Strong grasp of probability, statistics, and ML fundamentals; ability to distinguish real effects from noise and bugs using experimental data.
  • •Experience with curation, preprocessing, and analysis of large-scale text, code, or multimodal datasets.
Experience:AIMachine learningData engineeringResearch
Education:Bachelor's
Skills:CommunicationCollaborationProblem-solvingAnalytical thinking
Tech Stack:PythonPyTorchTensorFlowJAX

Eligibility

Work Authorization:Sponsorship available.

Company Brief

Thinking Machines Lab
Develops enterprise AI solutions, custom large language models, and ML platforms to help organizations deploy intelligent applications. Services include data engineering, model development, and AI consulting for scale and production readiness.
Industry: AI & Machine Learning
Company Size: Medium (51 to 250 employees)
Growth: Growth Stage Startup
Headquarters: Mumbai, India
Website