Staff ML Engineer

Role overview Lead the engineering of production AI systems that automate large-scale healthcare document and claims review. The work spans document processing, multimodal language-model workflows, reliability, and explainability, with the goal of supporting accurate payment-integrity decisions. The role also sets technical direction and helps other engineers build dependable AI systems.

Responsibilities - Build and scale pipelines for processing large annual volumes of medical records, claims, and benefits documents. - Design agent-based workflows to review and validate healthcare claims at high volume. - Improve the AI workflow ecosystem, including prompt development, multimodal text and vision models, context extraction, retrieval-augmented generation, orchestration, monitoring, logging, and output explanations. - Work with data science, data engineering, and product teams to turn requirements into end-to-end production systems. - Reduce technical debt and strengthen system scalability, reliability, and operational visibility. - Mentor engineers and guide practices for production-grade AI development.

Requirements - Deep software engineering experience, including meaningful work building enterprise AI-driven workflows or agentic systems. - Strong Java and Python skills; Scala experience is a plus. - Experience designing distributed systems and large-scale data pipelines, ideally involving unstructured documents or healthcare records. - A record of deploying AI/ML approaches such as prompt engineering, retrieval-augmented generation, workflow automation, or agents, with attention to reliability and explainability. - Familiarity with monitoring and observability tools such as Prometheus or Grafana, and strong computer science fundamentals in data structures, asynchronous programming, and system design. - Experience shipping enterprise AI/ML systems, with sound testing and code-quality practices; a relevant degree or equivalent practical experience.

Nice to have Experience with big-data and distributed-processing tools such as Spark, Flink, Kafka, Airflow, or Ray; experience automating work in healthcare, financial technology, or another regulated field is also relevant.

Benefits and work setup The position is digital-first and open to remote work anywhere in the United States. Listed benefits include medical, dental, and vision coverage; FSA/HSA options; tuition reimbursement; a 401(k) match; equity; wellness benefits; and unlimited paid time off. The posting gives a base-pay range of $180,000–$250,000 or more, depending on assessed level and candidate qualifications.