> Markdown version of [/jobs/ext/4352773-ml-engineer-infrastructure](https://www.wearedevelopers.com/jobs/ext/4352773-ml-engineer-infrastructure). Every page supports `.md` or `Accept: text/markdown`. Links point to the HTML versions so they work for humans too. Agent guide: [/agents.md](https://www.wearedevelopers.com/agents.md).

---

# ML Engineer - Infrastructure

- **Company:** FLEXION INC
- **Location:** San Francisco, CA, United States
- **Experience:** Expert
- **Contract:** Permanent contract
- **Skills:** Artificial Intelligence, Amazon Web Services, Microsoft Azure, Computer Clusters, Configuration Management, Profiling, Software Quality, File Systems, Distributed Computing Environment, Monitoring of Systems, Job Scheduling, Python (Programming Language), Machine Learning, Ansible, Software Engineering, Google Cloud, Cloud Platform System, Pytorch, Reliability of Systems, Scikit Learn, Kubernetes, Information Technology, Slurm, Machine Learning Operations, Hardware Infrastructure, Terraform
- **Published:** October 7, 2026
- **Apply:** https://www.thejobnetwork.com/job/96bbe95a-d655-4563-bfda-e014fd728dd6/ml-engineer-infrastructure

## About the Role

* Degree in Computer Science, Electrical Engineering or Software Engineering (or equivalent practical experience) plus significant industry experience.

 * Hands-on experience with the training or inference of large models (billions of parameters) on distributed multi-node GPU hardware. This can include bringing up and running the cluster, writing and optimizing training/inference code, building ML pipelines, etc.
 * Proficiency in Python and working knowledge of PyTorch.
 * Deep understanding of distributed training concepts (DDP, FSDP, NCCL).
 * Experience with at least one cloud platform (AWS, GCP, Azure or neoclouds) or large-scale on-premises GPU infrastructure.
 * Experience with job scheduling and orchestration tools: Slurm and/or Kubernetes/KubeRay.

Nice-to-haves

 * Familiarity with profilers (e.g., PyTorch Profiler, Dynolog, HTA, Nsight).
 * Experience with high-performance or parallel file systems (e.g., Lustre).
 * Experience provisioning compute on multiple cloud providers.
 * Experience with infrastructure-as-code and configuration management (Terraform, Ansible).

## Description

We are looking for an experienced ML engineer to join Flexion's experienced infrastructure team and take ownership of Flexion's GPU compute platforms. This is a senior, on-site role with significant scope.

At Flexion, we are building the brain for humanoid robots, which involves training foundation models with vast amounts of data on large GPU clusters. You will own the design, bring-up, operation and optimization of performant clusters. You will work with AI engineers to help them optimize their training speed and hardware utilization. You will also influence strategic compute planning and contribute to new tools and platforms for iterating on our AI models efficiently. This will put you at the heart of Flexion's AI development and allow you to directly impact the execution of our ambitious roadmap. You will closely collaborate with the company's leadership, engineers of the infrastructure team and AI engineers across the company., * Architect, run and continuously improve existing and future cloud-based GPU clusters. Select the best frameworks and tooling to run our clusters efficiently. Work on cluster provisioning, job schedulers and monitoring systems.

 * Help AI engineers optimize their training workloads and maximize hardware utilization using profilers, contributing to our core ML libraries.
 * Contribute to short- and long-term GPU compute strategies in collaboration with our AI engineering teams and help execute on them.
 * Optimize capacity and cost by exploring multi-cloud strategies and evaluating trade-offs.
 * Raise the bar on engineering practices, including testing, code quality, documentation, and system reliability.