Skip to content
Ambience HealthcareAmbience HealthcareSan Francisco, CA

Staff ML Engineer, Frontier AI

Staff ML Engineer owns foundational model research and end-to-end quality improvements for clinical AI products like coding models, adaptive scribing, and chart understanding. Requires 5+ years ML experience with deep RL/deep learning expertise and production shipping track record.

250k – 350k/yr
Hybrid5+ YOEML Engineering

About the role

What You’ll Own

  • Own foundational model research. Identify failure modes, form hypotheses, and drive architecture decisions on hard clinical AI problems — medical coding, adaptive scribing, chart understanding, and more.
  • Build compounding learning loops. Design systems that turn real-world signals — clinician edits, coder corrections, audit outcomes — into fast, safe model improvements.
  • Improve Chart Chat quality. Drive better grounding, smarter retrieval, and reasoning that holds up under the real diversity of clinical questions over complex longitudinal patient records.
  • Push latency, accuracy, and cost simultaneously. Apply the right optimization levers — capability routing, distillation, speculative decoding, quantization — and know when each is safe.
  • Contribute to population-level clinical reasoning. Help build toward a layer of clinical intelligence that reasons not just over individual patients, but across patient populations at scale.
  • Stay at the cutting edge. Distill insights from recent research — particularly in RL, deep learning, and clinical NLP — and drive experiments that keep Ambience at the frontier of clinical AI.

Who You Are

Deep RL and Deep Learning Expertise

  • 5+ years of ML engineering or applied research experience, with a strong track record of shipping model improvements in production.
  • Deep expertise in reinforcement learning and deep learning, developed in industry or a research setting.
  • Publications at top-tier venues (NeurIPS, ICML, ICLR, ACL, EMNLP, etc.) are a strong plus.

Research to Production

  • Comfortable spanning research and engineering — architecture decisions, training runs, fine-tuning pipelines, and production deployment.
  • Experience with preference learning, RLHF, retrieval-augmented generation, or multi-label classification.
  • Strong Python fundamentals and experience with deep learning frameworks (PyTorch preferred).

End-to-End Ownership

  • Can point to model quality improvements driven end to end: from identifying a failure mode to shipping and measuring a fix.
  • Has operated at the frontier of a hard problem, not just applied standard techniques.
  • Staff-level scope — has owned research directions and influenced technical decisions across teams.

Mission-Aligned

  • Passion for healthcare or other high-stakes, mission-driven industries.
  • Thrives in a fast-paced environment; takes extreme ownership of deliverables.

Nice-to-haves

  • Experience with clinical data: EHR systems, FHIR, medical coding ontologies, or clinical NLP.
  • Prior work in healthcare AI or other regulated, high-stakes domains.
  • Open-source contributions to ML libraries, benchmarks, or evaluation frameworks.

Compensation

Base compensation range of approximately $250,000-350,000 per year, exclusive of equity. Flexible cash and equity mix based on preferences. Comprehensive medical, dental, vision; 401(k) with 3% match; parental leave.

Skills

PyTorchPythonReinforcement LearningDeep LearningRLHFRetrieval-Augmented GenerationFine-TuningClinical NlpEhrFHIR

Similar roles

ML Engineering jobs
Axion

Staff Software Engineer, Agentic Platform

AxionSan Francisco, CA +1

Senior individual contributor architecting and scaling agentic LLM systems that turn messy manufacturing data into reliable root-cause insights. Owns orchestration, retrieval, evaluation, and guardrails for non-deterministic production systems.

250k – 270k/yr
Hybrid7+ YOEML Engineering
The Voleon Group

Member of Research Staff, Optimization

The Voleon GroupBerkeley, CA +1

Conduct optimization research and implement large-scale constrained optimization models that drive real-time trading decisions, working across the full research lifecycle from theory to production. Requires PhD-level coursework and strong applied research background in optimization.

250k – 275k/yr
Hybrid7+ YOEML Engineering
Nuance Labs

Member of Technical Staff — RL Research

Nuance LabsSeattle, WA

New/recent PhD to own RL and post-training for large-scale omni models. Build and scale the full RL/post-training stack including rollout, optimization, reward modeling, and evaluation for real-time audiovisual AI.

250k – 350k/yr
On-siteEntry levelML Engineering
Labelbox

Staff Software Engineer, AI Data Platform

LabelboxSan Francisco, CA

Staff-level engineer building AI data platform infrastructure, eval systems, and agent-first tooling for frontier labs. Requires 4+ years shipping production systems, full-stack experience, and deep TypeScript/Python proficiency.

250k – 280k/yr
Hybrid4+ YOEML Engineering
Nuance Labs

Member of Technical Staff — Model Optimization and Inference

Nuance LabsSeattle, WA

Optimize inference for real-time multimodal AI avatars. Specialize in LLM and diffusion model serving, KV cache strategies, quantization, and low-latency frameworks like vLLM and TensorRT-LLM.

250k – 350k/yr
On-site7+ YOEML Engineering