Software Engineer, Supercomputing

Thinking Machines Lab is seeking a Software Engineer to join our Supercomputing team in San Francisco. Our mission is to empower humanity by advancing collaborative general intelligence, providing access to AI tools tailored to individual needs. Our team comprises scientists and engineers who have contributed to widely used AI products and open-source projects.

In this role, you will design, build, and operate GPU supercomputing environments that support large-scale training and inference. Responsibilities include operating and automating large GPU clusters, developing software to abstract cluster management, extending scheduling systems for efficient resource allocation, monitoring operational metrics, and building reliable storage solutions. Collaboration with researchers to optimize performance and scalability is also key.

Candidates should have a Bachelor's degree or equivalent experience in computer science or a related field, proficiency in at least one backend language (such as Python or Rust), experience with large-scale clusters and container orchestration systems like Kubernetes or Slurm, and the ability to manage projects end-to-end. Preferred qualifications include a strong systems background, familiarity with CUDA/NCCL, experience supporting large-scale model training environments, understanding of deep learning frameworks, and a track record in fast-paced settings.

The position offers an annual salary between $350,000 and $475,000 USD, depending on experience. Benefits include generous health, dental, and vision coverage, unlimited PTO, paid parental leave, and relocation support as needed. Visa sponsorship is available for qualified candidates.

Thinking Machines Lab fosters a collaborative and innovative culture, encouraging team members to take initiative and work across different stacks and teams. We value diversity and are committed to providing equal employment opportunities to all applicants.