Distributed Production Inference Engineer

Thinking Machines Lab Inc.

San Francisco (CA)

On-site

USD 350,000 - 500,000

Full time

7 days ago
Be an early applicant

Get a reply from this employer

Your resume and cover letter, tailored to exactly what they’re hiring for.

Get past ATS filters

Benefits offered by this job

Health, dental, and vision benefits
Unlimited PTO
Paid parental leave
Relocation support

Job summary

Thinking Machines Lab Inc. is hiring a Software Engineer to design, build, and scale distributed production inference systems for Inkling, Inkling-Small, and Tinker.

You will own the end-to-end serving stack, from request routing to multi-node coordination and GPU utilization at scale. The role emphasizes production-first systems work, latency optimization, and deep observability, collaborating with research and infrastructure teams to translate evolving model architectures into reliable

Qualifications

  • 3+ years building and operating distributed systems in production.
  • Strong systems programming skills in Python, C++, Rust, or similar.
  • Experience with production infrastructure at scale: reliability, observability, and performance under real-world load.
  • Solid understanding of networking, concurrency, and distributed systems fundamentals.

Responsibilities

  • 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

Distributed systems
Python
C++
Rust
Networking
Observability
Concurrency

Tools

Kubernetes
CUDA
TensorRT
vLLM

Job description

Thinking Machines Lab Inc. is hiring a Software Engineer to design, build, and scale distributed production inference systems for Inkling, Inkling-Small, and Tinker.

You will own the end-to-end serving stack, from request routing to multi-node coordination and GPU utilization at scale. The role emphasizes production-first systems work, latency optimization, and deep observability, collaborating with research and infrastructure teams to translate evolving model architectures into reliable

Get your free, confidential resume review.

or drag and drop your file here.

Similar jobs

Similar jobs worth comparing

Distributed Inference Systems Engineer
Distributed Inference Systems Engineer

Speedrun Talent Network • San Francisco (CA)

On-site
USD 350,000 - 500,000
Health, dental, vision benefits
Unlimited PTO
Parental leave
+1
Software Engineer, Production Inference (Distributed Inference)
Software Engineer, Production Inference (Distributed Inference)

Speedrun Talent Network • San Francisco (CA)

On-site
USD 350,000 - 500,000
Health, dental, vision benefits
Unlimited PTO
Parental leave
+1
Software Engineer, Production Inference (Distributed Inference)
Software Engineer, Production Inference (Distributed Inference)

Thinking Machines Lab Inc. • San Francisco (CA)

On-site
USD 350,000 - 500,000
Health, dental, and vision benefits
Unlimited PTO
Paid parental leave
+1
Inference Systems Engineer — Scalable, Reliable ML Serving
Inference Systems Engineer — Scalable, Reliable ML Serving

Speedrun Talent Network • San Francisco (CA)

On-site
USD 300,000 - 400,000
Health, dental, and vision benefits
Unlimited PTO
Paid parental leave
+1
Software Engineer, Inference
Software Engineer, Inference

Speedrun Talent Network • San Francisco (CA)

On-site
USD 300,000 - 400,000
Health, dental, and vision benefits
Unlimited PTO
Paid parental leave
+1
Staff Inference Engineer — Production-Scale AI Platform
Staff Inference Engineer — Production-Scale AI Platform

Designworks Talent LLC • Bellevue (KY)

Hybrid
USD 180,000 - 240,000
Health insurance
401(k) plan with company match
Paid holidays
INFERENCE ENGINEER
INFERENCE ENGINEER

MakerMaker.AI • San Francisco (CA)

On-site
USD 120,000 - 160,000
Senior Inference Engineer, AI Infrastructure & Production
Senior Inference Engineer, AI Infrastructure & Production

Hamilton Barnes • United States

On-site
USD 225,000 - 275,000
Full Benefits
Senior Inference Engineer - Production Focus (Remote)
Senior Inference Engineer - Production Focus (Remote)

Vcluster • San Francisco (CA)

Hybrid
USD 190,000 - 220,000
Equity
Health, dental, vision, and life保险
Flexible working schedule
+1
Distributed Systems Engineer — AI Inference Platform
Distributed Systems Engineer — AI Inference Platform

Baseten • New York (NY)

On-site
USD 180,000 - 360,000
Competitive compensation with equity
Medical, dental, and vision coverage
Flexible PTO with Winter Break
+4