Latest AI JobsBrowse Jobs
D

AI Training Infrastructure Engineer

designworkstalentBellevueEngineering
Apply on company site
Salary
Not listed
Job type
Full-time
Work mode
Remote
Experience
Senior
Posted
7 Oct 2026
Status
Open · checked 7d ago

About this role

Based in a hybrid setup in downtown Bellevue, Washington, the AI Training Infrastructure Engineer position at designworkstalent involves building and scaling distributed systems for large-scale artificial intelligence models. This opportunity suits engineers who want to focus on distributed computing, GPU clusters, and reliability within a lean, collaborative team environment.

Role Overview

The organization is a well-funded, rapidly growing artificial intelligence infrastructure company building a cloud platform designed for the full lifecycle of AI. This foundational engineering role concentrates on the infrastructure layer behind large-scale model training. Professionals in this position collaborate with infrastructure, orchestration, performance, and machine learning teams to solve challenges involving distributed computing, fault tolerance, and training efficiency.

Key Responsibilities

Build and scale distributed training infrastructure supporting large artificial intelligence models across large GPU clusters.
Design and improve systems that increase training reliability, efficiency, and resource utilization.
Develop solutions for fault tolerance, checkpointing, recovery, and large-scale training operations.
Integrate models into production training pipelines in partnership with platform, orchestration, and performance engineering teams.
Diagnose and resolve issues impacting training throughput, stability, reliability, and cost efficiency.
Build tools and automation that improve the developer experience for researchers and engineers.
Establish best practices for training infrastructure, operational processes, and platform reliability.
Contribute to the evolution of the platform as an early member of the engineering team.

Requirements

Hands-on experience building and operating distributed training systems or large-scale machine learning infrastructure.
Experience supporting large artificial intelligence models, foundation models, post-training workflows, or similar systems.
Strong understanding of reliability, scalability, and efficiency challenges associated with multi-node GPU training.
Experience integrating training systems with production machine learning pipelines.
Strong programming skills and experience working with complex distributed systems.
Ability to independently own technically challenging projects in a fast-moving engineering environment.
Comfortable operating with high ownership and limited process overhead.

Nice to Have

Experience with distributed training frameworks such as PyTorch Distributed, DeepSpeed, Megatron-LM, Ray, or similar technologies.
Experience with supervised fine-tuning, reinforcement learning from human feedback, or other post-training workflows.
Background operating artificial intelligence training infrastructure at scale within a hyperscaler, research organization, cloud provider, or GPU cloud environment.
Experience optimizing GPU utilization, training performance, or distributed system reliability.
Familiarity with Kubernetes, containerized workloads, and large-scale infrastructure platforms.

Skills and Tools

PyTorch Distributed
DeepSpeed
Megatron-LM
Ray
Kubernetes
GPU clusters

Salary and Benefits

Competitive base pay for the Bellevue market.
Eligible roles may receive merit increases, annual bonuses, and long-term incentives based on individual performance.
United States-based employees receive medical, dental, and vision insurance.
Access to a 401(k) plan with a company match.
Paid holidays per calendar year.

About the Company

designworkstalent is recruiting for a well-funded, rapidly growing infrastructure company developing a next-generation cloud platform to support artificial intelligence workloads. The business combines the speed and innovation of a startup with the stability of an established parent organization, operating with intentional, lean, and AI-native engineering teams.

How to Apply

Interested candidates should apply on the official company careers page.

Skills and tech stack

PyTorch DistributedDeepSpeedMegatron-LMRayKubernetesPyTorchReinforcement LearningMachine LearningFine-tuningDistributed Systems

Questions candidates ask

What is the location of the role?

The role is hybrid and based in downtown Bellevue, Washington, requiring approximately three days per week in the office. Candidates elsewhere in the United States who are open to relocation may also apply.

Is visa sponsorship available for this position?

No, U.S. work authorization is required and visa sponsorship is not currently available.

What experience level is required?

The open roles include Senior and Staff titles, requiring hands-on experience with distributed training systems or large-scale machine learning infrastructure.

What benefits are provided for U.S. employees?

U.S.-based employees have access to medical, dental, and vision insurance, a 401(k) plan with company match, and paid holidays per calendar year.

What are the working arrangements regarding the office?

It is a hybrid position that requires working from the downtown Bellevue, Washington office approximately three days per week.

Ready to apply?

The application is on the official designworkstalent careers page.

Apply on company site