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.
![]()
Photo: Salesforce Tower via Wikimedia Commons, licensed CC BY-SA.
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
- Visa sponsorship confirmed
- High salary range
- Relocation support available
- 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.
Browse more USA Jobs and Visa Sponsorship Jobs on CMAJobs.








Join the discussion 0
No comments yet. Be the first to share your thoughts.