Staff ML Systems Engineer: Distributed AI for Robotics

Lever, Inc.; Seattle (WA); On-site; USD 195,000 - 230,000; Full time

Benefits offered by this job

Equity participation

Comprehensive benefits

Job summary

FieldAI is hiring a Staff ML Systems Engineer in Irvine to architect and build the distributed infrastructure powering large-scale ML workflows across the organization. You will design scalable systems for data processing, model training, evaluation, and post-processing, collaborating with ML and data teams to productionize research workflows and scale deployments.

You will define architectural patterns, tooling, and infrastructure, improving observability and performance in CPU/GPU

Qualifications

• 5+ years of experience building distributed systems or ML platforms.
• Experience with data pipelines and ML workflows.
• Strong Python programming and concurrent systems knowledge.
• Experience diagnosing bottlenecks in distributed environments.
• Ability to drive technical decisions across teams.

Responsibilities

• Design and build scalable distributed ML pipelines across data processing, model training, evaluation, and post-processing workflows.
• Architect distributed execution systems including scheduling, resource allocation, and fault tolerance.
• Develop reusable abstractions and libraries for distributed pipeline development.
• Optimize performance across CPU and GPU environments.
• Design data partitioning, memory usage, serialization overhead, and compute efficiency.
• Collaborate with ML engineers, data engineers, and infrastructure teams to productionize research workflows.
• Establish engineering standards for distributed ML infrastructure.
• Evaluate frameworks and trade-offs for distributed computing technologies.
• Improve observability, debugging, monitoring, and tooling for distributed systems.

Skills

Distributed systems

Python

Concurrency

Performance optimization

System design

Cross-functional collaboration

Tools

Kubernetes

Ray

Spark

PyTorch

TensorFlow

DeepSpeed

Job description

FieldAI is hiring a Staff ML Systems Engineer in Irvine to architect and build the distributed infrastructure powering large-scale ML workflows across the organization. You will design scalable systems for data processing, model training, evaluation, and post-processing, collaborating with ML and data teams to productionize research workflows and scale deployments.

You will define architectural patterns, tooling, and infrastructure, improving observability and performance in CPU/GPU