ML Infrastructure Engineer: Training & Real-Time Inference

Mach9 Robotics Inc.; San Francisco (CA); On-site; USD 180,000 - 240,000; Full time

Job summary

Mach9 Robotics Inc. is seeking an ML infrastructure engineer to build and maintain systems powering production AI models for civil engineering and surveying. You will manage training pipelines, data generation, and inference that integrates with CAD software.

The role targets mid-career engineers experienced with training and inference, focusing on scalable, reliable pipelines and real-time serving capabilities for field deployment.

Qualifications

• 3+ years of work experience in ML infra or related fields.
• Bachelor's or Master's degree in Computer Science, Engineering, or equivalent experience.
• Strong communication with ML researchers/engineers to translate workflows into robust systems.
• Experience designing data/versioning/artifact management or dataset lineage systems (e.g., DVC, LakeFS, Weights & Biases).
• Hands-on experience with ML pipeline orchestration tools (Airflow, Prefect, Metaflow).
• Experience with model serving and inference optimization (latency, memory footprint, scaling).
• Ability to read and refactor ML training code to improve reliability.
• Proficiency in Python and PyTorch.

Responsibilities

• Design and build a centralized system for versioning training data, datasets, and model artifacts with full lineage.
• Develop and maintain reliable ML training and data generation pipelines.
• Refactor training/data scripts into modular, testable components.
• Create CI/CD workflows for validating data pipelines and training runs with automated checks.
• Build tooling to launch, monitor, and debug training jobs with minimal friction.
• Optimize and scale real-time inference services to meet latency/throughput goals.
• Own deployment path from trained model to production endpoint with reliable rollouts and monitoring.

Skills

3+ years exp

Communication

Python

PyTorch

Team collaboration

Data versioning

ML pipeline tools

Model serving

Code readability

Education

BSc/MS in CS/Engineering

Tools

Airflow

Prefect

Metaflow

DVC

LakeFS

Weights & Biases

Terraform

Job description

Mach9 Robotics Inc. is seeking an ML infrastructure engineer to build and maintain systems powering production AI models for civil engineering and surveying. You will manage training pipelines, data generation, and inference that integrates with CAD software.

The role targets mid-career engineers experienced with training and inference, focusing on scalable, reliable pipelines and real-time serving capabilities for field deployment.