Advertisement

Research Engineer, Infrastructure, Inference at Thinking Machines Lab, San Francisco

cma9ja2 min read30

Looking for Research Engineer jobs in San Francisco with visa sponsorship? Thinking Machines Lab is hiring a Research Engineer, Infrastructure, Inference to design and scale systems that power large AI models.

Advertisement

Research Engineer Infrastructure Inference jobs in San Francisco with visa sponsorship - Salesforce Tower skyline

Photo: Salesforce Tower via Wikimedia Commons, licensed CC BY-SA.

Advertisement

Research Engineer Infrastructure Inference Jobs in San Francisco: Overview

Job Title Research Engineer, Infrastructure, Inference
Employer Thinking Machines Lab
Location San Francisco
Salary $400,000–$500,000 per year
Job Type Full-time, On-site
Experience Bachelor’s degree or equivalent
Visa Sponsorship Confirmed
Closing Date Not specified in the available job information

About the Employer

Thinking Machines Lab builds AI that extends human will and judgment. It trains frontier models, develops tools for people to make models their own, and creates interfaces for human-AI communication.

Job Responsibilities

  • Design, optimize and scale systems that power large AI models
  • Collaborate with research teams to enable high-performance inference
  • Implement techniques that improve performance, latency, throughput and efficiency
  • Optimize codebase and compute fleet to fully utilize hardware
  • Extend orchestration frameworks for distributed inference and evaluation
  • Establish standards for reliability, observability and reproducibility

Requirements

  • Bachelor’s degree or equivalent experience in computer science or engineering
  • Understanding of deep learning frameworks such as PyTorch or JAX
  • Experience with inference serving systems such as SGLang or vLLM
  • Strong engineering skills and ability to debug complex codebases
  • Preferred: experience with large-scale language models or open-source ML infrastructure

Visa Sponsorship and International Applicants

Thinking Machines Lab sponsors visas. While success is not guaranteed for every candidate, the company states it is committed to working through the visa process for the right fit.

Salary and Benefits

  • Generous health, dental and vision benefits
  • Unlimited PTO
  • Paid parental leave
  • Relocation support as needed
What you need to Know
Pros
  • Visa sponsorship confirmed
  • High salary range
  • Relocation support available
Cons
  • On-site in San Francisco
  • Evergreen role with no fixed closing date

How to Apply

Applications are submitted through the Thinking Machines Lab Ashby careers portal. A CV is required.

Apply to this Job

Advertisement

Browse more USA Jobs and Visa Sponsorship Jobs on CMAJobs.

Advertisement

Join the discussion 0

No comments yet. Be the first to share your thoughts.

Leave a comment