Skip to main content

Software Engineer, Distributed Systems

Thinking Machines LabSoftware
Pay
$300K–$350K
per year
Work mode
On-site
Full Time
Experience
5+ yrs
Mid

San Francisco, CA at a glance

Rent
#2 of 51
$2,680/mo+46% vs US avg
Weather
#17 of 51
295 mild days0 hot · 0 cold
Income tax
#1 of 51
13.3% top rateCalifornia

What you need

  • 5+ yrs building large-scale distributed systems
  • Proficiency in Python, Go, C++, or systems language
  • Deep understanding of consensus, consistency, replication, fault tolerance
  • Experience with network programming, load balancing, distributed storage

What you'll do

  • Design and build distributed orchestration, scheduling, storage, networking for GPU/TPU clusters
  • Develop fault-tolerant systems surviving hardware failures, network partitions, scale
  • Build distributed storage and data orchestration for large training/model data
  • Optimize collective communication, scheduling, resource allocation across thousands of machines
  • Partner with research/infra teams to identify bottlenecks and design first-principles solutions

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, Distributed Systems to design and build the core distributed systems that everything else at Thinking Machines runs on — orchestration, scheduling, storage, and networking across thousands of machines. Your work underpins both Inkling's training clusters and Tinker's serving platform, and shows up anywhere we need software to coordinate reliably at scale.

This is deep systems work. You'll be reasoning about consensus, fault tolerance, and performance under real-world failure conditions, often on problems that don't have an off-the-shelf solution.


What You'll Do

  • Design and build distributed systems for compute orchestration, scheduling, storage, and networking across large GPU and TPU clusters

  • Develop fault-tolerant systems that keep running correctly as hardware fails, networks partition, and workloads scale

  • Build the distributed storage and data orchestration layers that move and persist large volumes of training and model data

  • Improve the performance and efficiency of collective communication, scheduling, and resource allocation across thousands of machines

  • Partner with research and infrastructure teams to identify systems bottlenecks and design solutions from first principles

  • Write production-quality code and help shape the architecture of systems used company-wide


Skills & Qualifications

Minimum Qualifications

  • 5+ years of experience building large-scale distributed systems

  • Proficiency in Python and Go, C++, or another systems-level language

  • Strong understanding of distributed systems fundamentals: consensus, consistency, replication, and fault tolerance

  • Experience with network programming, load balancing, or distributed storage systems

Preferred Qualifications

  • Experience building distributed compute or orchestration systems for AI or ML workloads

  • Fluent in containerization, orchestration, and distributed compute frameworks

  • Experience with specialized hardware (GPUs, TPUs) and their integration into distributed training or serving systems

  • Background at AI research labs, high-performance computing centers, or similarly demanding environments

  • Published work or open-source contributions related to distributed systems or performance engineering

  • Comfortable operating with high autonomy in a fast-changing, early-stage environment


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 $300,000 - $350,000 USD.

  • 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.

Optimize your resume for this job

Get a match score and the keywords you're missing

Optimize resume

About 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.

Similar Software roles

Software Engineer, Distributed Systems
$300K–$350K · Thinking Machines Lab
Apply