ML Infrastructure Engineer

About this role

Mach9 is seeking a mid-career ML Infrastructure Engineer to develop and manage systems for AI models in civil engineering and surveying. The role involves building training pipelines for large datasets and architecting inference infrastructure for seamless integration with CAD software.

What you'll do

• Design a centralized system for versioning training data and model artifacts
• Develop and maintain reliable ML training and data generation pipelines
• Refactor existing training scripts into maintainable components
• Create CI/CD workflows for data validation and model training
• Build tools for ML engineers to manage training jobs
• Optimize real-time model inference services for production

What they're looking for

• 3+ years of experience in ML infrastructure
• Strong communication skills
• Experience with data versioning and artifact management
• Hands-on with ML pipeline orchestration tools
• Model serving and inference optimization
• Ability to read and refactor ML code
• Proficient in Python and PyTorch