Thinking Machines Lab

Thinking Machines Lab

Software Engineer, Supercomputing

San Francisco

H1B sponsorship available$350k-$475kDetected 110 days ago
PythonRustExpressKubernetesLinuxDeep LearningTensorFlowPyTorchLogistics

About the role

  • Note: This is an "evergreen role" that we keep open on an on-going basis to express interest.
  • We receive many applications, and there may not always be an immediate role that aligns perfectly with your experience and skills.
  • We continuously review applications and reach out to applicants as new opportunities open.

Responsibilities

  • Build reliable storage and artifact paths for datasets, checkpoints, and logs with clear retention and lineage.
  • Partner with researchers to unblock scale runs and advise on parallelism and performance trade‑offs.
  • Thrive in a highly collaborative environment involving many, different cross-functional partners and subject matter experts.

Requirements

  • Bachelor's degree or equivalent experience in computer science, engineering, or similar.
  • Proficiency in at least one backend language (we use Python or Rust).
  • Experience operating large‑scale clusters and container orchestration systems (e.g. Kubernetes or Slurm).
  • Comfort operating across the stack and owning projects end-to-end.
  • Familiarity with CUDA/NCCL and performance profiling for distributed training/inference.
  • Track record of working in fast-paced environments balancing care with urgency.

Nice to have

  • we encourage you to apply if you meet some but not all of these:

Skills

  • Thinking Machines Lab's mission is to empower humanity through advancing collaborative general intelligence.

Compensation

  • Depending on background, skills and experience, the expected annual salary range for this position is $350,000 - $475,000 USD.

Benefits

  • Thinking Machines offers generous health, dental, and vision benefits, unlimited PTO, paid parental leave, and relocation support as needed.

Visa & Work Authorization

  • 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.
  • We sponsor visas.

This listing is sourced directly from Thinking Machines Lab's careers page and normalized into a canonical job model.