T

Infrastructure Research Engineer

thinking machines lab • United State
Visa Sponsorship Relocation
Apply Now

Design and build core systems for scalable and efficient training of large-scale AI models. Optimize distributed training stacks to maximize GPU throughput and ensure research reliability. Requires strong systems engineering skills and proficiency with deep learning frameworks.

Key Highlights
Scale distributed training systems across thousands of GPUs and nodes
Optimize training stack end-to-end to eliminate system bottlenecks
Develop reusable frameworks to improve reproducibility and scalability
Key Responsibilities
Design, implement, and optimize distributed training systems that scale across thousands of GPUs and nodes for large-scale training workloads.
Develop high-performance optimizations to maximize throughput and efficiency.
Develop reusable frameworks and libraries to improve training reproducibility, reliability, and scalability for new model architectures.
Establish standards for reliability, maintainability, and security, ensuring systems are robust under rapid iteration.
Collaborate with researchers and engineers to build scalable infrastructure.
Publish and share learnings through internal documentation, open-source libraries, or technical reports.
Technical Skills Required
Distributed Training Deep Learning Frameworks Systems Engineering
Benefits & Perks
Health, dental, and vision benefits
Unlimited PTO
Paid parental leave
Relocation support
Visa sponsorship
Nice to Have
Past experience working on distributed training for the world’s largest models
Track record of improving research productivity through infrastructure design or process improvements
Contributions to open-source ML infrastructure such as PyTorch, XLA, Megatron-LM, or DeepSpeed

Job Description

About Thinking Machines

The mission of Thinking Machines is to build AI that extends human will and judgment. We are training frontier models with Inkling, developing Tinker to let people make models their own, and crafting interfaces that broaden human-AI communication. We believe the future worth building is human, and we're hiring people who want to build it.

About the Role

Want the full job description? Read the complete details on LinkedIn, the original posting.
Continue on LinkedIn

This is a short excerpt. All rights to the full description belong to its original publisher.

See Jaabz jobs first on Google 1 tap · free · in AI Overviews Jaabz is on your Google Manage Preferred Sources

Similar Jobs

Explore other opportunities that match your interests

Visa Sponsorship Relocation Remote
Job Type Full-time
Experience Level Not Applicable

wayve

United State

Junior AI/ML Engineer

Machine Learning
•
6h ago
Visa Sponsorship Relocation Remote
Job Type Full-time
Experience Level Entry level

Tech Consulting

United State

AI/ML Engineer

Machine Learning
•
6h ago
Visa Sponsorship Relocation Remote
Job Type Full-time
Experience Level Entry level

Tech Consulting

United State

Subscribe our newsletter

New Things Will Always Update Regularly