Skip to content
Thinking Machines LabThinking Machines LabSan Francisco, CA

Research, Vision Expertise

Conducts research on visual perception, multimodal learning, and large-scale AI model training. Designs architectures, builds datasets and evaluations, and collaborates on frontier models. Requires ML expertise, Python proficiency, and experimental rigor.

350k – 475k
On-siteML Engineering

About the role

What You’ll Do

  • Own research projects on training and performance analysis of multimodal AI models.
  • Curate and build large-scale datasets and evaluation benchmarks to advance vision capabilities.
  • Work with our data infrastructure engineers, pretraining researchers and engineers, and product team to create frontier multimodal models and the products that leverage them.
  • 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

Minimum qualifications:

  • Ability to design, run, and analyze experiments thoughtfully, with demonstrated research judgment and empirical rigor.
  • Understanding of machine learning fundamentals, large-scale training, and distributed compute 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.

Preferred qualifications:

  • Research or engineering contributions in visual reasoning, spatial understanding, or multimodal architecture design.
  • Experience developing evaluation frameworks for multimodal tasks.
  • Publications or open-source contributions in vision-language modeling, video understanding, or multimodal AI.
  • A strong grasp of probability, statistics, and ML fundamentals. You can look at experimental data and distinguish between real effects, noise, and bugs.
  • PhD in Computer Science, Machine Learning, Physics, Mathematics, or a related discipline with strong theoretical and empirical grounding; or, equivalent industry research experience.

Logistics

Compensation: Depending on background, skills and experience, the expected annual salary range for this position is $350,000 - $475,000 USD.

Benefits: Generous health, dental, and vision benefits, unlimited PTO, paid parental leave, and relocation support as needed.

Skills

PyTorchTensorFlowJAXPythonMachine LearningMultimodal AiVision-Language ModelsDistributed TrainingLarge-Scale DatasetsEvaluation Frameworks

Similar roles

ML Engineering jobs
Thinking Machines Lab

Research Infrastructure Engineer

Thinking Machines LabSan Francisco, CA

Build and operate research infrastructure like evaluation frameworks, RL training systems, and experiment tracking platforms. Partner directly with ML researchers to identify bottlenecks, ensure high adoption of tools, and accelerate research velocity.

350k – 475k
On-siteML Engineering
Anthropic

Research Engineer, Safeguards Labs

AnthropicSan Francisco, CA +1

Research engineer on the Safeguards Labs team building and evaluating novel safety methods to detect misuse, strengthen model safeguards, and reduce real-world harm from Claude.

350k – 850k
HybridML Engineering
Anthropic

Research Engineer, Knowledge Team

AnthropicSan Francisco, CA +2

Designs new information architectures for LLMs to interact with external data sources, implements finetuning/RL training, builds evaluation sets, and develops agentic search capabilities. Requires strong Python/ML skills and LLM experience.

350k – 850k
HybridML Engineering
Thinking Machines Lab

Research, Pre-Training Data

Thinking Machines LabSan Francisco, CA

Designs and implements methods for sourcing, curating, and analyzing large-scale pre-training datasets for AI models, blending research with production-grade data engineering. Requires Python proficiency, deep learning frameworks, and strong ML fundamentals.

350k – 475k
On-siteML Engineering
Thinking Machines Lab

Research, Post-Training Data

Thinking Machines LabSan Francisco, CA

Conducts post-training research for AI models, designing data collection strategies, developing labeling pipelines, modeling human preferences, and iterating on evaluations to improve model alignment, reasoning, and helpfulness. Requires strong Python skills, ML framework proficiency, and experimental rigor.

350k – 475k
On-siteML Engineering