Remote Senior AI Engineer — Healthcare Document Automation

AgentGraph; United States; Hybrid; USD 150.000 - 190.000; Vollzeit

Benefits dieser Stelle

401(k) match

Company HSA contributions

Wellness reimbursement

Unlimited PTO

Parental leave

Mental health resources

Annual bonus

Equity opportunities

In-person company kick-off

Zusammenfassung

AgentGraph is seeking a Senior Applied AI Engineer to design, build, and operate AI-driven services for healthcare document processing. You will shape the architecture of intelligent processing, including LLM-powered document understanding, extraction, retrieval, and multi-step workflows.

You will own production systems, mentor engineers, and collaborate with Product Management, System Architects, DevOps, and Customer Success.

Qualifikationen

• Substantial production experience building LLM-powered systems that real users depend on.
• Depth in document understanding and structured extraction, including OCR and layout analysis of varied sources.
• Evidence-driven practice: you build evaluation before tuning and show measurable outcomes.
• Judgment about approach: knowing when to fine-tune, use retrieval, or classical ML.
• Comfort owning production systems, debugging under pressure, and balancing cost and latency.

Aufgaben

• Design, develop, and maintain LLM-powered services and pipelines for document understanding and extraction.
• Build agentic, multi-step workflows with routing, validation, and human-in-the-loop review.
• Choose approaches (hosted frontier models, fine-tuned models, or classical ML) based on quality, latency, and cost.
• Create and maintain quality datasets and labeling strategies for ground-truth ambiguity.
• Collaborate with product management and QA to meet customer needs and reliability standards.
• Build evaluation suites and regression detection to ship with evidence over intuition.
• Establish observability with tracing, cost accounting, and drift monitoring.
• Lead code reviews focused on ML/LLM, data processing, and production readiness.

Kenntnisse

LLM-powered systems

Document understanding

Production-grade ML

Jobbeschreibung

Build the agentic systems modernizing healthcare document workflows — extraction, routing, and multi-step automation over clinical and administrative documents.

Compensation

The base pay offered may vary depending on experience, qualifications, and location.

AgentGraph is running this confidential search on behalf of a client: a healthcare technology company modernizing how clinical and administrative documents move between providers, payers, and patients. We share the client's identity and the full team context with candidates early in the process.

Why this role exists

Healthcare still runs on documents. Referrals, prior authorizations, clinical summaries, lab results, intake forms, and claims move as faxes, scans, and PDFs — unstructured, inconsistently formatted, and processed by people doing work that software should have absorbed a decade ago. The cost shows up as staff hours, delayed care, and errors that surface later in the chart or the bill.

This role owns the AI systems that change that. You will design, build, and operate the LLM-powered services that read these documents, extract what matters, route them where they need to go, and drive the multi-step workflows around them — with the accuracy and auditability healthcare demands. It is a rare combination: genuinely hard applied AI problems, on a document corpus that punishes naive approaches, where getting it right removes real friction from how care gets delivered and paid for.

The Senior Applied AI Engineer designs, builds, and operates the AI-driven services that provide core functionality to the platform. You will shape the technical architecture of the intelligent processing and machine learning systems, with a focus on LLM-powered document understanding, structured extraction, retrieval pipelines, and agentic workflows.

You will operate with genuine autonomy, treating technical decisions as product decisions, and taking ownership from ambiguous problem statements to measured production outcomes. You will collaborate closely with Product Management, System Architects, DevOps, and Customer Success, serving as a technical lead who mentors less experienced engineers and engages directly with customers on complex solution design or escalations.

Team: Engineering · Reports to: VP of Engineering · Direct reports: None / individual contributor

What you will do

• Design, develop, and maintain the LLM-powered services and pipelines at the core of the platform — document understanding, structured data extraction, classification, and summarization across highly variable real-world clinical and administrative documents
• Build agentic and multi-step workflows where they demonstrably outperform simpler approaches — routing, validation, exception handling, and human-in-the-loop review — with guardrails and fallbacks appropriate to a clinical setting
• Select the right approach for each problem — hosted frontier models, fine-tuned open-weight models, or classical ML — based on quality, latency, and cost, and implement fine-tuning where it demonstrably beats prompting and retrieval on those measures
• Create and maintain the high-quality datasets that model development, fine-tuning, retrieval, and evaluation depend on, including labeling strategy for documents where ground truth is genuinely ambiguous
• Partner with Product Management and Quality Assurance so AI solutions meet real customer needs and performance requirements, following the company's software development life cycle and its reliability, scalability, and reproducibility standards
• Build and maintain evaluation suites for LLM-powered features — curated golden datasets, automated scoring calibrated against human review, and regression detection — so that model, prompt, and pipeline changes ship with evidence rather than intuition
• Establish observability for production AI services, including request tracing, token and cost accounting, and monitoring for quality drift and failure modes such as malformed or unfounded model output. In healthcare, a confidently wrong extraction is the failure mode that matters most, and the systems you build should surface it rather than pass it downstream
• Lead code reviews with particular attention to ML- and LLM-specific concerns such as data processing, prompt and model versioning, experiment tracking, evaluation coverage, and production readiness

Problem solving & optimization

• Provide problem-solving support for ML and AI systems, including performance optimization, debugging model and retrieval behavior, and improving pipeline efficiency
• Optimize inference for production environments — latency, throughput, token and compute cost, batching, caching, model routing, and model right-sizing — treating inference cost as a product constraint, not an afterthought
• Optimize ML pipelines for scalability, latency, and resource utilization at real production document volumes
• Serve as a technical point of contact for escalated customer issues involving ML and AI services, partnering with Customer Success to own each issue end to end — reproduce, root-cause, fix, and close the loop with affected stakeholders
• Participate directly in customer-facing engagements when required — solution design sessions, technical discussions, and integration support — representing the team's capabilities credibly to technical and non-technical audiences, and translating customer needs into engineering direction
• Feed lessons from customer engagements back into product, dataset, and system improvements

Technical thought leadership & mentoring

• Drive complex projects from ambiguous problem statements through design, implementation, and measured rollout, actively identifying and pursuing high-impact opportunities that align with company strategy
• Influence the architecture of ML/LLM systems and mentor engineers by sharing best practices, guiding implementation efforts, and providing constructive feedback that raises the team's overall technical standard

What we look for

• Substantial production experience building LLM-powered systems that real users depend on — not prototypes, demos, or notebooks that never met a difficult document
• Depth in document understanding and structured extraction, and a realistic view of what OCR, layout analysis, and scanned or faxed source material actually do to a pipeline
• An evidence-driven practice: you build the evaluation before you tune, and you can show how a change moved a measured outcome
• Judgment about approach — knowing when a fine-tune, a retrieval pipeline, or plain classical ML beats reaching for the largest available model
• Comfort owning production systems, including debugging under pressure and treating cost and latency as things you are accountable for
• The ability to handle regulated data responsibly. Experience working under HIPAA with PHI is a strong plus; a willingness to work carefully within those constraints is a requirement
• Enough customer-facing ability to be trusted in a solution design session with a customer's technical and clinical stakeholders

Key relationships

Internal: Customer Success, DevOps, Product Management, Quality Assurance, System Architects

External: Customer technical and operational stakeholders — engaged as needed for escalations, solution design, and technical integration discussions, typically in partnership with Customer Success and implementation teams

About the engineering department

The Engineering department designs, develops, and maintains the company's software products, covering development, infrastructure, operations, quality assurance, and product management. Its objective is to build and sustain products that satisfy customer needs and align with the long-term business strategy, to research and integrate new technologies that improve internal products and operations, and to ensure everything it ships meets the company's IT, information security, and compliance requirements.

Location and work authorization

Remote-first — open to candidates located anywhere in the United States.

Applicants must be legally authorized to work in the United States without employer sponsorship, now or in the future. The client does not sponsor, transfer, or take over sponsorship of employment visas for this role, and cannot consider candidates who require sponsorship at any point.

Customer-facing scope

This role has the ability to be up to 25% customer-facing. The work is primarily engineering, but the successful candidate should be comfortable stepping into customer-facing solution design, technical discussions, and escalation support as needed.

$160K – $200K base salary, plus equity. The base pay offered may vary depending on the candidate's experience, qualifications, and location. The company also offers an annual bonus opportunity, evaluated together with base and equity as a total compensation package based on the candidate's experience, qualifications, and overall fit.

• An incredible 401(k) match program to help you invest in your future
• Company HSA contributions to support your healthcare savings
• A monthly health and wellness reimbursement for the things that help you feel your best
• Flexible, unlimited PTO and a remote-first environment that supports life outside of work
• Competitive paid parental leave, mental health resources, and Employee Assistance Program support
• Annual bonus opportunities based on role and company performance
• Equity opportunities that allow you to share in the company's growth and success
• An annual in-person company kick-off where the team connects, collaborates, and celebrates together
• The opportunity to make a meaningful impact at a small, fast-growing technology company with a supportive, inclusive, and genuinely fun culture

The company welcomes applicants from all backgrounds, especially those underrepresented in technology. It does not discriminate based on ancestry, race, place of origin, political belief, religion, marital or family status, physical or mental ability, sex, sexual orientation, gender identity or expression, age, or any other legally protected characteristic. The company is an equal opportunity employer.

How To Apply

Send Us Your Application

with the following information:

• In the Subject: The role you are applying for — “Senior Applied AI Engineer - Healthcare AI”
• Resume/CV
• A portfolio showcasing your work (e.g., GitHub Account)

Using something else? Copy the address above and email us from there.

If you don't see anything that fits your background or aspirations, but you're wanting to work at AgentGraph, email us a description of what you'd like to do here + your resume.