ML Platform Engineer

Hadrian Automation • Los Angeles, CA

1d • $170,000 - $300,000

About The Position

This is an ML infrastructure role at the core of Hadrian’s technology stack. While Data Science, Operations Research, Vision, and Document AI teams build models, you will own the platform that ensures these models remain reliable, effective, and secure in production. You’ll standardize our deployment patterns built around MLflow, Dagster, ECR, FastAPI, and EKS, making them the backbone for packaging, evaluating, releasing, serving, monitoring, and rolling back models across Hadrian’s automated factories.

Requirements

• Track record building and operating production ML infrastructure across multiple models or inference workloads.
• Strong production-level Python and SQL skills, including typing, testing, packaging, API design, and building 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.
• Practical judgment around latency, throughput, availability, multi-tenancy, autoscaling, and infrastructure cost optimizations.
• Ability to build stable interfaces and collaborate closely with engineering and scientific stakeholders.

Nice To Haves

• Experience implementing feature stores (Feast, Tecton, or internal systems).
• Production work with Ray Serve, KServe, Triton, BentoML, SageMaker, Vertex AI, or custom gRPC inference services.
• Experience serving and evaluating vision, document-understanding, embedding, or generative pipelines.
• Expertise in GPU inference optimization, multi-model serving, edge inference, or Go/Rust performance-sensitive AI services.
• Background in regulated environments or open-source contributions to ML infrastructure projects (MLflow, Feast, KServe, Ray).

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.

Benefits

• Medical, dental, vision, and life insurance
• 401(k)
• Flexible vacation policy
• Relocation support may be provided for certain situations, based on business need.
• Equity

What This Job Offers

• Job Type: Full-time
• Career Level: Senior
• Education Level: No Education Listed