ML Infra Engineer — Real-Time AI & Data Pipelines

Mach9; San Francisco (CA); On-site; USD 150,000 - 210,000; Full time

Job summary

Mach9 is seeking a mid-career ML infrastructure engineer to design and operate scalable ML training and deployment systems for production AI models used in civil engineering and surveying. You will build end-to-end pipelines for large 3D point cloud and image data, and architect inference services that integrate with CAD software to deliver real-time results.

You will work with Python and PyTorch, implement data lineage and artifact management, and develop CI/CD workflows to ensure

Qualifications

• 3+ years of work experience in relevant fields.
• Bachelor's or Master's degree in Computer Science, Engineering, or equivalent experience.
• Strong communication skills and the ability to work closely with ML researchers and engineers to understand their workflows and translate them into robust systems.
• Experience designing and building data versioning, artifact management, or dataset lineage systems (e.g., DVC, LakeFS, Weights & Biases, or custom solutions).
• Hands-on experience with ML pipeline orchestration tools (e.g., Airflow, Prefect, Metaflow, or similar).
• Experience with model serving and inference optimization — profiling latency, reducing memory footprint, or scaling serving infrastructure to meet real-time constraints.
• Ability to read and refactor ML training code — you don't need to design model architectures, but you need to understand what training pipelines are doing well enough to make them reliable.
• Proficient with Python, PyTorch.

Responsibilities

• Design and build a centralized system for versioning training data, generated datasets, and model artifacts, with full lineage tracking from raw source data through to trained model outputs.
• Develop and maintain reliable, reproducible ML training and data generation pipelines.
• Refactor and harden existing training and data generation scripts into composable, testable, and maintainable components.
• Create CI/CD workflows for validating data pipelines and model training runs, including automated correctness checks and regression detection.
• Build tooling that enables ML engineers to launch, monitor, and debug training jobs with minimal friction.
• Optimize and scale real-time model inference services to meet latency and throughput requirements in production, including profiling, batching strategies, and resource-efficient serving.
• Own the deployment path from trained model artifact to production endpoint, ensuring reliable rollouts, rollback, and monitoring.

Skills

Python Proficiency

PyTorch

Data versioning

ML pipelines

CI/CD

Communication

Transformer models

Education

Bachelor's or Master's in CS/Engineering

Tools

Airflow

Prefect

Metaflow

DVC

LakeFS

Weights & Biases

AWS

Terraform

Docker

Job description

Mach9 is seeking a mid-career ML infrastructure engineer to design and operate scalable ML training and deployment systems for production AI models used in civil engineering and surveying. You will build end-to-end pipelines for large 3D point cloud and image data, and architect inference services that integrate with CAD software to deliver real-time results.

You will work with Python and PyTorch, implement data lineage and artifact management, and develop CI/CD workflows to ensure