Staff ML Engineer, AI Platform jobs in United States
cer-icon
Apply on Employer Site
company-logo

Ambience Healthcare · 1 day ago

Staff ML Engineer, AI Platform

Ambience Healthcare is the leading AI platform for documentation, coding, and clinical workflow, built to reduce administrative burden and protect revenue integrity at the point of care. The role involves building evaluation and release infrastructure, debug tooling, and chart context retrieval layers to enhance AI performance across health systems.

Artificial Intelligence (AI)Health CareInformation TechnologyMedical
check
H1B Sponsor Likelynote

Responsibilities

Eval & Release Infrastructure — Automated graders and release gates that work across product pods. Unified eval dataset versioning and execution to replace fragmented workflows. Production quality monitoring with end-to-end tracing, shared metrics, and automated alerting
Debug Tooling — Encounter replay that reconstructs exact inference inputs (retrieved chart context, packed prompts, model versions) so teams reproduce issues without digging through logs. Diff views comparing known-good runs to regressions
Chart Context & Data Pipelines — The retrieval layer that pulls relevant patient history and assembles it into consistent model-ready inputs. Feedback loops that capture real-world usage and convert it into training signal. End-to-end latency instrumentation across every workflow step
Preference Infrastructure — The system that enables clinician and site-specific behavior across specialties. Different clinics want different defaults, different phrasing, different workflows. You'll build the platform that supports customization at scale
Model Serving — Performance and reliability layer for critical in-house models with clear SLOs, capacity planning, and regression alerts

Qualification

ML infrastructureData pipelinesPythonTypeScriptMLOpsBackend fundamentalsPlatform engineeringEval systemsML observability

Required

7+ years in software engineering, 3+ focused on ML infrastructure, platform engineering, or data systems
Staff-level scope: owned cross-cutting infrastructure, influenced technical direction across multiple teams
Strong backend fundamentals in Python, TypeScript, or similar
Built eval systems, data pipelines, or ML observability infrastructure
Comfortable on both the ML and Eng sides of MLOps
Track record of platform work that measurably accelerated other teams
In SF, 3x/week in-person

Benefits

Health, dental, and vision coverage
Quarterly retreats
Unlimited PTO
401(k) plan with matching

Company

Ambience Healthcare

twittertwitter
company-logo
Ambience Healthcare develops an AI operating system used for documentation, coding, and clinical workflows in healthcare organizations.

H1B Sponsorship

Ambience Healthcare has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (4)

Funding

Current Stage
Growth Stage
Total Funding
$319.29M
2025-07-29Series C· $243M
2024-02-06Series B· $70M
2022-04-25Series A

Leadership Team

leader-logo
Nikhil Buduma
Co-Founder & CEO
linkedin
leader-logo
Michael Ng
President & Chairman
linkedin
Company data provided by crunchbase