Other open roles
Member of Technical Staff, Model Training (infra & platform)
Work Expert (WE) is an independent publishing and referral website. We are not a recruiter, hiring manager, agent or employer, and we are not affiliated with or endorsed by micro1. Applying takes you to the platform's own website, where we may be recorded as the referring source. We may receive a referral fee at no additional cost to you. Read our full affiliate disclosure →
About the Role
The Role In this role, you will take technical ownership of the distributed systems and platforms that enable researchers to train increasingly capable models, execute large-scale experiments, and deploy specialized AI systems efficiently. You will work across distributed training, GPU infrastructure, inference optimization, simulation environments, and agent execution frameworks, making critical architectural decisions that directly influence research velocity and model performance.
What You'll Do
- Architect, build, and scale shared infrastructure for supervised fine-tuning (SFT), reinforcement learning (RL), and on-policy distillation (OPD), supporting language, multimodal, and robotics workloads.
- Design distributed training and execution pipelines that coordinate GPU workers, rollout generation, teacher-model inference, reward computation, and simulation environments.
- Develop high-performance infrastructure for model training and inference, optimizing GPU utilization, memory management, communication, storage, and distributed execution.
- Build and evolve the core architecture of micro1's agent ecosystem, including execution runtimes, tool interfaces, environment integrations, model registries, and serving APIs.
- Develop robust systems for experiment tracking, dataset and model versioning, checkpointing, fault recovery, and observability to improve research reproducibility and operational reliability.
- Identify and eliminate infrastructure bottlenecks, reducing experiment turnaround time and enabling researchers to run increasingly complex training workloads.
- Define technical standards, platform interfaces, and architectural patterns that support collaboration across research, infrastructure, and applied engineering teams.
- Lead critical engineering initiatives from system design through implementation and production operations, mentoring engineers while maintaining direct ownership of technically challenging components.
You're a Good Fit If You
- Required skills: Distributed ML Training Infrastructure, GPU Performance & Inference Optimization, Distributed Systems & Platform Architecture, Technical Leadership & System Ownership
Role Highlights
Pay & Payout
Engaged directly through micro1. Apply via the link below - micro1 handles onboarding and payment.
Common Questions
Opens micro1 in a new tab. You pay nothing; we may earn a referral fee.
Get similar jobs every week
About micro1 for this role
We track 368 open micro1 roles on AI Work Expert. This role lists $600K-$1.8M/yr. Our reliability rating for micro1 is reliable, with an onboarding time of 1-3 days.
Similar roles
More coding work you may qualify for.
Member of Technical Staff, Model Training (applied research)
The Role In this role, you will work across language models, multimodal systems, robotics, and autonomous agents, developing models optimized for specific applications and environm
Member of Technical Staff, Model Training (post-training research)
The Role In this role, you will investigate how learning algorithms, training data, and evaluation methodologies interact to produce stronger reasoning, generalization, and agent c
Member of Technical Staff, Data Engineering
The Role This is a Staff-level, hands-on engineering role for someone who has built data infrastructure from zero to production and knows how to turn fragmented operational data in