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

Ambience Healthcare · 23 hours ago

Staff ML Engineer, AI Platform

Ambience Healthcare is the leading AI platform for documentation, coding, and clinical workflow, aimed at reducing administrative burden in healthcare. The Staff ML Engineer will build and maintain infrastructure to enhance AI capabilities across health systems, focusing on evaluation, release processes, and data pipelines.
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 engineeringDebug toolingEval systemsObservability infrastructure

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
A 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
Key Investors
Andreessen Horowitz,Oak HC/FTKleiner Perkins,OpenAI Startup Fund
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