GrepJob
Mach9

ML Infrastructure Engineer

Mach9
Apply
2 months ago
San Francisco, CA, USAMid Level / Senior
H1B Sponsor

Base Salary

$160k - $200k/yr

Responsibilities

  • Design and build 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 composable and maintainable components.
  • Create CI/CD workflows for validating data pipelines and model training.
  • Build tooling for ML engineers to launch and monitor training jobs.
  • Optimize real-time model inference services for production requirements.
  • Own the deployment path from trained model artifact to production endpoint.

Requirements

  • 3+ years of work experience in relevant fields.
  • Bachelor's or Master's degree in Computer Science, Engineering, or equivalent experience.
  • Strong communication skills to collaborate with ML researchers and engineers.
  • Experience designing data versioning and artifact management systems.
  • Hands-on experience with ML pipeline orchestration tools.
  • Experience with model serving and inference optimization.
  • Ability to read and refactor ML training code.
  • Proficient with Python and PyTorch.

Tech Stack

Apache AirflowAWSDVCPythonPyTorchTerraform

Categories

AI & MLData EngineeringDevOps