ML Platform Engineer — Production AI Infra

Hadrian Automation; Los Angeles (CA); On-site; USD 170,000 - 300,000; Full time

Benefits offered by this job

Medical, dental, vision, and life insu

401(k)

Flexible vacation policy

Equity

Job summary

Hadrian is building autonomous factories in the United States, seeking an ML infrastructure lead to own the production platform around ML models. You will standardize deployment patterns with MLflow, Dagster, ECR, FastAPI, and EKS, serving diverse workloads and ensuring reliability, security, and observability across Hadrian’s factories.

You’ll drive release and evaluation processes, manage online feature serving, and collaborate with engineering and scientific stakeholders to optimize latency,

Qualifications

• Strong production-level Python and SQL skills, including typing, testing, packaging, API design, and observability features.
• Hands-on experience with Kubernetes, containers, and handling distributed-system failure modes such as retries, partial failures, idempotence, and resource isolation.
• Engineering background with model registries, feature systems, batch/real-time inference, experiment tracking, or model CI/CD workflows.

Responsibilities

• Build the production platform that enables Hadrian’s factories to safely depend on models for drawing extraction, cycle-time prediction, forecasting, scheduling, and more—with measurable performance and fast rollback.
• Develop shared batch and online serving for tabular, vision, document-AI, scheduling, graph, and embedding workloads, targeting clear SLAs for latency, availability, and isolation.
• Create repeatable release and evaluation processes featuring automated tests, reproducible artifacts, lineage, shadow deployments, canaries, and A/B tests.
• Own online feature serving and maintain contract integrity with offline feature tables; proactively detect and address training-serving skew, feature drift, bad data, and model degradation.
• Build operational tooling for telemetry, incident response, autoscaling, resource and GPU management, cost attribution, and secure model routing.
• Develop APIs, SDKs, reusable templates, and documentation that teams can adopt without requiring close support from platform engineers.

Skills

Python

SQL

Kubernetes

APIs

Observability

Tools

MLflow

Dagster

ECR

FastAPI

EKS

Job description

Hadrian is building autonomous factories in the United States, seeking an ML infrastructure lead to own the production platform around ML models. You will standardize deployment patterns with MLflow, Dagster, ECR, FastAPI, and EKS, serving diverse workloads and ensuring reliability, security, and observability across Hadrian’s factories.

You’ll drive release and evaluation processes, manage online feature serving, and collaborate with engineering and scientific stakeholders to optimize latency,