Z

Zus Health

Subscription platform integrating patient health data

Staff/Principal AI/ML Engineer

Applied Machine Learning +1Confirmed live in the last 24 hours
Full-time

About the Job

Requirements

  • At least 7 years of experience building and shipping AI/ML systems, including at least 2 years putting LLM-powered products into production for real users.
  • Hands-on depth in modern LLM engineering, including retrieval-augmented generation, agents and tool use, structured outputs, prompt and context engineering, and trade-offs between frontier APIs and open-weight models.
  • A track record building ML infrastructure used by others, including serving, pipelines, model and prompt versioning, continuous integration and delivery, and monitoring.
  • Experience designing evaluation sets and metrics and applying a rigorous approach to evaluation.
  • Strong Python and software engineering fundamentals, with fluency in PyTorch and the Hugging Face ecosystem.
  • Production experience on a major cloud, preferably AWS, with containers and infrastructure as code.
  • Sound judgment about safety, privacy, and reliability when models handle sensitive data.
  • Clear writing and communication skills to explain model behavior and risk to clinicians, product leaders, and executives.
  • A BS in Computer Science, Statistics, or a related field, or equivalent experience.
  • Must be in the Boston metro for hybrid in-person collaboration, or located remotely in the EST/CST time zones and willing to travel occasionally for in-person collaboration.

Responsibilities

  • Architect shared LLM platform infrastructure, including model gateway and routing across frontier and open-weight models, prompt and version management, caching, cost and latency controls, and PHI-safe guardrails.
  • Build tool-using agents that reason over longitudinal patient records to summarize histories, reconcile medications, surface care gaps, and answer clinician questions with cited evidence, using MCP servers, function calling, and multi-step planning.
  • Build retrieval-augmented generation and hybrid search over FHIR resources and unstructured notes, including chunking and embedding strategies for clinical text, vector and graph retrieval, reranking, and grounding answers in source documents.
  • Establish offline and online evaluations, including golden datasets with clinical reviewers, LLM-as-judge pipelines, regression suites in continuous integration, and production monitoring for hallucination, drift, and safety.
  • Fine-tune and distill models using LoRA/PEFT, supervised fine-tuning, and preference optimization for clinical extraction, coding, and normalization tasks.
  • Build continuous integration and delivery for models and prompts, inference serving, observability, and human-in-the-loop feedback loops that use user corrections to improve models.
  • Work directly with providers, product managers, and engineers to identify high-leverage problems, ship prototypes quickly, and iterate based on real usage.
  • Lead design reviews, write documentation and standards, mentor engineers, and keep the team current with developments in AI.

Desired Qualifications

  • Healthcare or clinical natural language processing experience, especially with FHIR, C-CDA, clinical terminologies, or HIPAA-regulated data.
  • Experience fine-tuning, distilling, or applying preference optimization to LLMs at meaningful scale.
  • Experience building MCP servers or multi-agent systems used in production.
  • Inference optimization experience, including quantization, speculative decoding, batching, or GPU cost management.
  • Experience designing and executing pre-training of transformer models.
  • Open-source contributions, publications, or talks in machine learning, natural language processing, or LLM systems.
  • Experience as the first or founding ML engineer at a growth-stage company.
  • An MS or PhD in a quantitative field.

About the Company

Zus Health

Subscription platform integrating patient health data

Zus Health builds a platform that plugs into healthcare workflows to give providers quick access to patients’ histories, medication adherence data, and other important health information. The product helps health systems, telehealth services, and chronic care teams by reducing manual data gathering and easing administrative tasks, while offering real-time alerts for medication pickups and streamlined patient intake. It differentiates itself by focusing on seamless integration, timely, actionable insights, and a holistic view of patient data to improve care delivery and efficiency, rather than reinventing the care process from scratch. The company’s goal is to improve patient outcomes and operational efficiency by making patient data more accessible and useful for clinicians and care teams.

Company size
51-200
Company stage
Series B
Total funding
$107M
Headquarters
Boston, MA
Founded
2020

Recently Posted Jobs

Explore more opportunities at Zus Health

See more jobs