Your browser does not support javascript! Please enable it, otherwise web will not work for you.

ML Infrastructure Engineer

Home > Python programming jobs

ML Infrastructure Engineer in USA new

  • San Francisco, CA

Responsibilities

  • Design and build systems for versioning training data, generated datasets, and model artifacts with end-to-end lineage tracking.
  • Develop reliable and reproducible ML training and data-generation pipelines.
  • Refactor and harden ML training and data-generation scripts into composable, testable, and maintainable components.
  • Create CI/CD workflows with automated correctness checks and regression detection for data pipelines and model training runs.
  • Build tooling for ML engineers to launch, monitor, and debug training jobs.
  • Optimize and scale real-time model inference services through latency profiling, batching, and resource-efficient serving.
  • Own reliable deployment, rollout, rollback, and monitoring processes from trained model artifacts to production endpoints.

Requirements

  • At least 3 years of relevant work experience.
  • Bachelor’s or master’s degree in Computer Science, Engineering, or equivalent experience.
  • Strong communication skills and ability to collaborate with ML researchers and engineers.
  • Experience building data versioning, artifact management, or dataset lineage systems such as DVC, LakeFS, Weights & Biases, or custom solutions.
  • Hands-on experience with ML pipeline orchestration tools such as Airflow, Prefect, or Metaflow.
  • Experience with model serving and inference optimization, including latency profiling, memory reduction, or serving infrastructure scaling.
  • Ability to read and refactor ML training code without necessarily designing model architectures.
  • Proficiency with Python and PyTorch.
  • Preferred experience with AWS infrastructure, containerized ML workflows, GPU-accelerated training, model optimization, infrastructure-as-code, and large unstructured datasets.

Salary: $160k - $250k/yr

Mach9

Mach9 develops AI-powered geospatial software that converts reality-capture datasets (LiDAR and imagery) into high-precision 3D maps and engineering-ready CAD/GIS deliverables. Its flagship product, Digital Surveyor, automatically extracts assets such as utility poles, signage, curbs, and stripin...

Similar positions

Senior Software Engineer

  • Mozilla
  • Full time
  • Remote
  • 10/04/2026
  • Salary: €66k-€88k
  • Netherlands

DevOps Engineer, Radar

  • Anduril Industries
  • Full time
  • USA
  • 10/04/2026
  • Salary: $111k - $147k/yr
  • Fort Collins, CO

Senior Software Engineer

  • Mozilla
  • Full time
  • Remote
  • 10/04/2026
  • Salary: €56k-€74k
  • Belgium

Senior Research Engineer

  • AssemblyAI
  • Full time
  • USA
  • 10/04/2026
  • Salary: $270k-$310k
  • Remote

Senior Product Engineer

  • Clera
  • Full time
  • Others
  • 10/04/2026
  • Berlin, Germany