Latest AI JobsBrowse Jobs
Thinking Machines logo

Research Engineer, Infrastructure, Training Systems

Thinking MachinesSan FranciscoResearch Infrastructure (ML Infrastructure and Training Stack)
Apply on company site
Salary
$350K - $475K USD
Job type
Full-time
Work mode
Remote
Posted
4 Aug 2026
Status
Open · checked 7d ago

About this role

Thinking Machines is seeking a Research Engineer for Infrastructure and Training Systems in San Francisco, offering a remote-friendly setup and a salary range of $350,000 to $475,000 per year. This position is ideal for professionals who combine systems and performance expertise with a passion for large-scale machine learning.

Role Overview

This position involves designing and building core systems that facilitate scalable and efficient training of large models for both deployment and research purposes. The primary goal is to ensure fast, reliable experimentation so that research teams can concentrate on science rather than system bottlenecks. Candidates will take end-to-end ownership of the training stack, maximizing the impact of every GPU cycle.

Key Responsibilities

Design, implement, and optimize distributed training systems that scale across thousands of nodes and GPUs for large workloads.
Create high-performance optimizations to drive maximum throughput and operational efficiency.
Build reusable libraries and frameworks to enhance training scalability, reproducibility, and reliability for modern model architectures.
Set standards for security, maintainability, and reliability to keep systems robust during rapid iteration.
Partner closely with engineers and researchers to develop scalable infrastructure solutions.
Share insights and learnings via open-source libraries, technical reports, or internal documentation to advance scalable AI infrastructure.

Requirements

Hold a Bachelor’s degree or possess equivalent experience in computer science, statistics, electrical engineering, physics, machine learning, robotics, or a related field.
Demonstrate strong engineering capabilities to write performant, maintainable code and debug complex codebases.
Possess a solid understanding of deep learning frameworks, including PyTorch, and their underlying system architectures.
Thrive in a collaborative environment alongside cross-functional partners and subject matter experts.
Display a bias for action and a proactive mindset to navigate multiple stacks and teams to ensure successful delivery.

Nice to Have

Previous experience working on distributed training for major global models to ensure stability, performance, and reliability.
A proven track record of boosting research productivity via process improvements or infrastructure design.
Past contributions to open-source machine learning infrastructure projects such as DeepSpeed, Megatron-LM, PyTorch, or XLA.

Skills and Tools

PyTorch
Distributed Training Systems
Deep Learning Framework Architectures
High-Performance Optimization Frameworks

Salary and Benefits

Annual salary range between $350,000 and $475,000 USD, determined by background, skills, and experience.
Comprehensive medical, dental, and vision health coverage.
Unlimited paid time off (PTO).
Paid parental leave.
Relocation assistance provided when necessary.
Visa sponsorship available for qualified candidates.

About the Company

Thinking Machines builds artificial intelligence designed to extend human will and judgment. The organization trains frontier models utilizing Inkling, creates Tinker to empower individuals to customize models, and crafts interfaces that expand human-AI communication.

How to Apply

Interested candidates should submit their application directly through the official Thinking Machines careers page.

Skills and tech stack

PyTorch

Questions candidates ask

What is the salary range for the Research Engineer position?

The expected annual salary range is between $350,000 and $475,000 USD, depending on the candidate's skills, background, and experience.

Is visa sponsorship available for this role?

Yes, Thinking Machines provides visa sponsorship and works with the candidate through the visa process.

Where is the job located?

The role is based in San Francisco, California, with remote flexibility.

What benefits are offered by Thinking Machines?

The company offers generous health, dental, and vision benefits, unlimited paid time off, paid parental leave, and relocation support as needed.

Is this an evergreen job posting?

Yes, this is an evergreen role kept open on an ongoing basis to collect expressions of interest, and applications are continuously reviewed.

Ready to apply?

The application is on the official Thinking Machines careers page.

Apply on company site