AI Infrastructure Generalist - Build Next-Gen Clusters

Thinking Machines Lab Inc.; San Francisco (CA); Hybrid; USD 350,000 - 475,000; Full time

Benefits offered by this job

Visa sponsorship

Relocation support

Health benefits

Job summary

Thinking Machines Lab Inc. in San Francisco, California seeks generalist infrastructure and systems engineers to build the foundations powering our AI models and to support research and product teams.

You will join a small, high-impact group responsible for architecting and scaling core infrastructure, solving distributed systems challenges, and delivering robust, scalable platforms. This role offers visa sponsorship, competitive compensation—$350,000 to $475,000 annually—and relocation support

Qualifications

• Bachelor’s degree or equivalent in CS/engineering or similar.
• Proficiency in Python or Rust as a backend language.
• Experience operating large-scale clusters and container orchestration (Kubernetes/Slurm).
• Comfort operating across the stack and owning projects end-to-end.
• Thrive in a collaborative, cross-functional environment.
• Bias for action and initiative to drive shipping across stacks.

Responsibilities

• Architect and scale core infrastructure powering AI models.
• Collaborate with researchers to accelerate experiments and improve efficiency.
• Build robust, scalable platforms across the full technology stack.
• Support data and research teams with reliable infrastructure and tooling.

Skills

Python

Rust

Kubernetes

End-to-end ownership

Collaboration

Education

Bachelor’s degree in CS/Engineering

Tools

Spark

Slurm

CI/CD

GPU workloads

Job description

Thinking Machines Lab Inc. in San Francisco, California seeks generalist infrastructure and systems engineers to build the foundations powering our AI models and to support research and product teams.

You will join a small, high-impact group responsible for architecting and scaling core infrastructure, solving distributed systems challenges, and delivering robust, scalable platforms. This role offers visa sponsorship, competitive compensation—$350,000 to $475,000 annually—and relocation support