At a glance
Machine Learning Infrastructure Engineer is an open full-time job at True Talent Solutions Partners (San Francisco · On-site).
| Company | True Talent Solutions Partners |
|---|---|
| Location | San Francisco · On-site |
| Type | Full-time |
| Department | Client Roles |
| Pay | Create a free account to see the pay |
| Posted | 2026-11-05 |
| Found on | Ashby |
About this job
About the Role
Join an AI research team building large-scale models for physical systems, with weather as an initial application. You will develop the infrastructure for distributed model training and inference, helping turn large volumes of observational data into useful predictions and insights.
What You'll Do
- Design, deploy, and maintain distributed machine learning training and inference clusters.
- Build scalable pipelines for petabyte-scale datasets and manage model training across the machine learning lifecycle.
- Test training approaches, including parallelization methods and numerical precision trade-offs across model sizes.
- Profile and debug low-level GPU operations to improve performance.
- Evaluate research and apply relevant techniques to infrastructure and training workflows.
- Analyze training experiments, including results from ablation studies, to inform model development.
What We're Looking For
- Two to ten years of experience building large-scale machine learning infrastructure for pretrained or foundation models.
- Hands-on experience with compute clusters, data infrastructure, inference systems, and distributed training.
- Strong skills in optimizing large-scale training, data, or inference workloads, including GPU performance debugging.
- Proficiency with distributed training frameworks such as FSDP or DeepSpeed, and experience with CUDA and JAX.
- Experience with Python, C++, Linux, NVIDIA GPUs, and cloud platforms such as GCP, AWS, or Azure.
- Familiarity with Docker, Kubernetes, distributed task management, model serving, monitoring, logging, and observability.
- Background in physical AI or AI for science, such as robotics, self-driving, or biology, and interest in mission-driven work.
- A generalist approach to the machine learning lifecycle and experience delivering in fast-paced, time-sensitive environments.
Compensation & Benefits
Visa sponsorship is not available. Relocation support is provided.
Location
On-site in San Francisco, California, United States, five days per week.
Questions about this job
Where is the Machine Learning Infrastructure Engineer job at True Talent Solutions Partners?
San Francisco.
What does the Machine Learning Infrastructure Engineer job at True Talent Solutions Partners pay?
The listing states a pay range. Create a free account on this site to see it.
Is this a full-time or part-time job?
The listing describes it as Full-time.
How do I apply to True Talent Solutions Partners?
Create a free job seeker account on this site, then use the Apply button to open the employer's own posting.
Hiring for this role at True Talent Solutions Partners? Create your hiring profile with your work email, so job seekers can find and connect with you.
All jobs at True Talent Solutions Partners · More jobs near San Francisco