T

Thinking Machines Lab

Researches AI and builds tools to customize models

Software Engineer - Inference

Applied Machine LearningConfirmed live in the last 24 hours
Full-timeSan Francisco, CA, USA$300k - $400k /yr

About the Job

Requirements

  • Experience operating large-scale, latency-sensitive production systems.
  • Proficiency in Python and Go or another systems language.
  • Experience with observability, monitoring, and incident response for production services.
  • Strong understanding of distributed systems and how they fail at scale.

Responsibilities

  • Operate and scale the production inference systems that serve live traffic, including Tinker's multi-tenant serving platform.
  • Own the rollout process for new models, model versions, and inference optimizations, ensuring safe, incremental deployment to production.
  • Build and improve observability, alerting, and capacity planning so the team can detect, diagnose, and resolve production issues quickly.
  • Partner with inference and research teams to productionize new serving techniques without compromising reliability.
  • Lead incident response for production inference issues, driving root cause analysis and durable fixes.
  • Design for graceful degradation, failover, and redundancy so that serving stays resilient as usage grows.
  • Manage capacity and cost tradeoffs for serving infrastructure as traffic and model sizes scale.

Desired Qualifications

  • Experience running production inference for large language models or other large-scale machine learning systems.
  • Experience with deployment and rollout systems, such as canarying, blue/green deployments, or feature flags.
  • Experience with capacity planning and cost optimization for graphics processing unit or tensor processing unit infrastructure.
  • Familiarity with inference-specific techniques, such as batching, caching, or quantization, and their operational implications.
  • Comfort being on-call and leading incident response for critical production systems.
  • Comfort working with high autonomy in a fast-changing, early-stage environment.

About the Company

Thinking Machines Lab

Researches AI and builds tools to customize models

Thinking Machines Lab researches and builds AI systems, aiming to make AI more widely understood, customizable, and capable. It publishes much of its research openly on its Connectionism blog, starting with work on making AI models give consistent answers to the same question. Its first product, Tinker, lets researchers and developers fine-tune open-weight AI models on their own data while Thinking Machines runs the computing clusters behind it, so customers focus on their data and training methods instead of managing hardware and pay for the computing they use. Former OpenAI chief technology officer Mira Murati founded the lab in 2025, and it is based in San Francisco.

Company size
201-500
Company stage
Series A
Total funding
$7B
Headquarters
San Francisco, CA
Founded
2024

Recently Posted Jobs

Explore more opportunities at Thinking Machines Lab

See more jobs