Software Engineer, Infrastructure at Thinking Machines Lab | CA, US | Rezi

Software Engineer, Infrastructure at Thinking Machines Lab

Software Engineer, Infrastructure

Thinking Machines Lab · CA, US

1 months ago

Software Engineer, Infrastructure

Thinking Machines Lab · CA, US

a month ago
Resume preview

Impress employers and recruiters.
Choose from hundreds of resume examples.

Target Resume Now
Resume preview

Tailor your resume to this Software Engineer, Infrastructure role.

Rezi rewrites your resume against Thinking Machines Lab's job description. Free.

Resume score gauge reading 58 out of 100

Don't guess if your resume is good enough.

See how it scores against the Software Engineer, Infrastructure posting at Thinking Machines Lab — free, in seconds.

About the Role

We're hiring a Software Engineer, Infrastructure to design and build the distributed systems that power our model training and serving platforms. You'll work on the systems underlying everything we do — from the clusters that train our frontier models with Inkling, to the multi-tenant serving infrastructure behind Tinker. This is a foundational infrastructure role at a fast-moving startup with real ownership over systems that run at large scale, and your work will directly determine how quickly our research and product teams can iterate.

Responsibilities

  • Design, build, and operate distributed systems that support large-scale model training and inference across thousands of accelerators
  • Build and maintain core infrastructure, including orchestration, scheduling, storage, and resource management systems
  • Improve the reliability, performance, and observability of infrastructure used across research and product teams
  • Partner with researchers and platform engineers to understand infrastructure needs and turn them into robust, well-abstracted systems
  • Debug and resolve complex distributed failures across the stack, from networking and storage to compute and scheduling
  • Write and maintain internal libraries and APIs, primarily in Python and Go, that other engineers build on

Requirements

  • Demonstrated expertise designing and developing large-scale distributed systems
  • Strong proficiency in Python and Go
  • Experience building, deploying, and operating production infrastructure at scale
  • Solid grounding in distributed systems fundamentals, such as consensus, consistency, fault tolerance, and networking
  • Experience with ML infrastructure, such as training orchestration, job schedulers, or distributed storage and data systems
  • Experience operating large-scale GPU or TPU clusters
  • Experience with container orchestration (e.g. Kubernetes) and infrastructure-as-code
  • Contributions to open-source infrastructure projects
  • Comfortable working with high autonomy in a fast-changing, early-stage environment

Skills

  • Python
  • Go
  • Distributed systems fundamentals
  • Consensus
  • Consistency
  • Fault tolerance
  • Networking
  • ML infrastructure
  • Training orchestration
  • Job schedulers
  • Distributed storage
  • Data systems
  • GPU clusters
  • TPU clusters
  • Kubernetes
  • Infrastructure-as-code

Location

  • San Francisco, CA

Work Type

  • Onsite

Experience Level

  • Senior

Salary/Compensations

  • $300,000 - $350,000 USD

Benefits

  • Generous health, dental, and vision benefits
  • Unlimited PTO
  • Paid parental leave
  • Relocation support

About the Company

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