Looking for research engineer jobs in San Francisco with visa sponsorship? Thinking Machines Lab is hiring a Research Engineer, Infrastructure, Training Systems to design and optimize distributed training infrastructure for large-scale AI models.
Research Engineer Jobs in San Francisco: Overview
| Job Title | Research Engineer, Infrastructure, Training Systems |
| Employer | Thinking Machines Lab |
| Location | San Francisco, CA |
| Salary | $350,000–$475,000 per year |
| Job Type | Full-time, On-site |
| Experience | Bachelor’s degree or equivalent in CS, EE, ML or similar; strong engineering skills |
| Visa Sponsorship | We sponsor visas. Committed to working through the process for the right fit. |
| Closing Date | Not specified in the available job information |
About the Employer
Thinking Machines Lab is an AI research company focused on advancing collaborative general intelligence. The team builds systems and tools to make large-scale model training and research efficient and reliable.
Job Responsibilities
- Design, implement and optimize distributed training infrastructure for large-scale AI models across thousands of GPUs and nodes
- Build high-performance frameworks and libraries that improve throughput, reliability, reproducibility and scalability
- Own the training stack end to end, ensuring experiments run fast and reliably
- Collaborate with researchers to translate algorithmic ideas into production-grade training pipelines
- Develop shared monitoring and observability tools for high uptime and debuggability
Requirements
- Bachelor’s degree or equivalent experience in computer science, electrical engineering, statistics, machine learning, physics, robotics or similar
- Strong engineering skills with ability to contribute performant, maintainable code and debug complex codebases
- Understanding of deep learning frameworks such as PyTorch or JAX and their system architectures
- Experience with large-scale distributed training preferred
Visa Sponsorship and International Applicants
Thinking Machines Lab sponsors visas. While success cannot be guaranteed for every candidate or role, the company is committed to working through the visa process together for the right fit. The role is based in San Francisco.
Salary and Benefits
- Salary $350,000–$475,000 per year depending on background and experience
- Health, dental and vision benefits
- Unlimited PTO
- Parental leave
- Relocation support
- High compensation package
- Direct impact on large-scale training systems
- Visa sponsorship available
- On-site in San Francisco required
- High technical bar for distributed systems
- Sponsorship success not guaranteed for all candidates
How to Apply
Applications are submitted via the company’s Ashby job board. Provide your resume and relevant experience details.
Application Tips
- Emphasize experience with distributed training, PyTorch/JAX and large-scale systems
- Prepare examples of optimizing throughput or reliability in production ML environments
Browse more Technology Jobs and USA Jobs on CMAJobs.
Join the discussion 0
No comments yet. Be the first to share your thoughts.