ML Infrastructure Engineer

Job description

The role At Mach9, ML infrastructure engineers build and maintain the systems that power production AI models for civil engineering and surveying. Our ML pipeline spans 10,000+ miles of labeled survey data, image segmentation networks, and 3D prediction models serving real-time inference to surveyors and engineers in the field. This role is ideal for mid-career ML infrastructure engineers with experience building for both training and inference. You'll build training pipelines that handle deep transformer models on hundreds of terabytes of 3D point cloud and image data. You'll also architect our inference infrastructure, delivering both heavy offline detection algorithms and real-time responsive inference that integrates directly with our CAD software. 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 workfl...