logo

AI Job Summary

Design and operate distributed production inference systems serving large language models, optimizing for latency, throughput, and cost at scale. You'll handle request routing, batching, multi-node serving, and GPU utilization while working with research and infrastructure teams. Requires 3+ years building distributed systems in production and strong systems programming in Python, C++, or Rust. Based in San Francisco. Salary $350,000–$500,000.

Written from this posting by Neural Jobs AI. The full description is below.

AI Resume Tailoring Sign in to use this AI Cover Letter Sign in to use this

Job Description

About Thinking Machines

The mission of Thinking Machines is to build AI that extends human will and judgment. We are training frontier models with Inkling, developing Tinker to let people make models their own, and crafting interfaces that broaden human-AI communication. We believe the future worth building is human, and we're hiring people who want to build it.

About the Role

We're hiring a Software Engineer to build and scale the distributed production inference systems that serve Inkling, Inkling-Small, and Tinker in production. You'll own the systems that turn trained models into fast, reliable, cost-efficient services — from request routing and batching to multi-node serving and GPU utilization at scale.

This is a systems-heavy, production-first role. You'll work closely with research and infrastructure teams to translate rapidly evolving model architectures into serving systems that meet real-world latency, throughput, and reliability requirements, and you'll be on the front line when production inference systems need to scale, recover, or improve.

What You'll Do

  • Design, build, and operate distributed infrastructure for large-scale model serving, including request routing, load balancing, batching, and multi-node coordination

  • Optimize inference latency and throughput in production, including work on KV cache management, continuous batching, speculative decoding, and quantization

  • Build and maintain high-concurrency serving systems with strong uptime, low tail latency, and deep observability

  • Benchmark, tune, and extend inference engines to support new model architectures as they move from research into production

  • Partner with research and infrastructure teams to translate emerging model designs into production-ready serving systems

  • Build tooling for tracing, debugging, and resolving issues across the serving stack, from orchestration down to GPU kernels

  • Participate in on-call rotation to support production inference systems

Skills & Qualifications

  • 3+ years of experience building and operating distributed systems in production

  • Strong systems programming skills in Python, C++, Rust, or similar languages

  • Experience with production infrastructure at scale: reliability, observability, and performance under real-world load

  • Solid understanding of networking, concurrency, and distributed systems fundamentals

Preferred Qualifications

  • Experience with LLM inference engines such as vLLM, SGLang, or TensorRT-LLM

  • Familiarity with GPU programming (CUDA) or low-level performance optimization

  • Experience with model parallelism, tensor/pipeline parallelism, or other distributed inference techniques

  • Track record of operating large-scale production systems with strict latency and uptime requirements

  • Experience with Kubernetes or similar orchestration systems for GPU workloads

  • Contributions to open-source ML systems or inference infrastructure projects

Logistics

  • Location: This role is based in San Francisco, CA.

  • Compensation: Depending on background, skills and experience, the expected annual salary range for this position is $350,000 - $500,000 USD (placeholder — verify against current internal bands before publishing).

  • Visa sponsorship: 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.

  • Benefits: Thinking Machines offers generous health, dental, and vision benefits, unlimited PTO, paid parental leave, and relocation support as needed.

Do you match this job?

Here is what this employer asked for. Sign in and we will fill in your half.

  • Role Software Engineer, Production Inference (Distributed Inference)
  • Experience 3-4 years
  • Work type On-site
  • Location United States
Check my match (free)
Thinking Machines Lab
AI Research Lab · 500+ Members · Nashville, TN, United States

Thinking Machines Lab is an AI research and product company developing advanced models and tools that can adapt to individual needs. Its work focuses on making powerful AI more understandable, customizable, multimodal, and useful for collaboration with people. The company combines frontier research with products for model use and customization, including tools that let developers work with open-weight models. Thinking Machines Lab states a broader goal of giving more people access to the knowledge and capabilities required to shape AI for their own applications.

404 more Software Engineer roles in San Francisco

Job Overview

Approx. salary range

192K – 287K

Our estimate — this employer did not publish a salary

Our estimate, not the employer’s. Worked out from the middle half of 50 comparable roles on Neural Jobs that did publish a salary, in the same field, country and experience band. The real figure for this job may be different.

Eligibility
United States Right to work in the United States required.
Workplace
On-site
Job Posted:
1 day ago
Job Type
Full Time
Experience
3-4 years

Share This Job: