← All roles

Research, Pre-Training Science

Thinking Machines Lab· Research
Apply on company site
San Francisco, CA$350k - $475kfull-timePosted 11mo ago

About the role

Research-focused role at Thinking Machines Lab centered on advancing pre-training methods, architectures, and learning objectives for large models, blending fundamental research with practical engineering.

Responsibilities

  • Research and develop new methodologies for pre-training.
  • Work in areas such as scaling, architecture, algorithms, or optimization of large scale training runs depending on your research interest and experience.
  • Design data curricula and sampling strategies that improve learning dynamics and model generalization.
  • Collaborate with infrastructure and data teams to conduct large-scale experiments efficiently and reproducibly.
  • Publish and present research that moves the entire community forward. Share code, datasets, and insights that accelerate progress across industry and academia.

Qualifications

  • Ability to design, run, and analyze experiments thoughtfully, with demonstrated research judgment and empirical rigor.
  • Experience with distributed or high-performance computing environments.
  • 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.

Skills

Required:Experiment design and analysisDistributed/high-performance computingPython programmingDeep learning frameworksCommunication

Benefits

  • generous health, dental, and vision benefits
  • unlimited PTO
  • paid parental leave
  • relocation support as needed
Last updated Jul 8, 2026. Interested? Apply on the company site.
Thinking Machines Lab logo

About Thinking Machines Lab

Connectionism: Research Blog by Thinking Machines Lab

Thinking Machines Lab is an artificial intelligence research and product company. We’re building a future where everyone has access to the knowledge and tools to make AI work for their unique needs and goals.

AIartificial intelligencebessemer-ai-infrastructure-2026deep learningliquid-ai-recruiting-targetmachine learningproductreinforcement-learning-platforms
Company profileWebsite

More open roles at Thinking Machines Lab