AI Engineer

Posted 10 Hours Ago
New York, NY, USA
Hybrid
145K-175K Annually
Mid level
Artificial Intelligence • Healthtech • Insurance • Machine Learning • Payments
Claims operations at the speed of AI.
The Role
Build and productionize AI agents for healthcare claims review. Responsibilities include agent orchestration, LLM fine-tuning, RAG pipelines, reinforcement learning alignment, evaluation frameworks, production integration, and HIPAA-compliant safety guardrails. The role requires Python, PyTorch, Hugging Face, production RAG experience, AI agent development, and applied reinforcement learning expertise.
Summary Generated by Built In

About Alaffia & Our Mission

Every year, U.S. health plans lose billions to improper payments and administrative waste. That wasted spending ultimately trickles down across the healthcare ecosystem, driving up costs for plans, providers, and patients alike. We’re here to change that paradigm.

Alaffia is a new kind of claims operations partner for health plans. Using expert clinicians and transparent AI, we deliver deeper insights, smarter automation, and consistently better outcomes across the entire lifecycle of claims. With Alaffia, health plans can cut wasted spending more effectively than ever — and provide their members the most affordable care.

We’re a high-growth, venture-backed Series B healthtech startup based in NYC and are actively scaling our company. Join us in helping to build a healthcare system that works better for everyone.

*This position requires current authorization to work in the United States. Unfortunately, we are not in a position to sponsor work visas at this time.

About the Role & What You’ll Be Doing

Alaffia Health is a healthcare AI startup revolutionizing medical billing and data automation. Our AI-driven platform leverages state-of-the-art generative AI and machine learning technologies to enhance accuracy, efficiency, and compliance in medical billing and auditing

  • As we scale, we’re looking for an AI Engineer to build the intelligence behind Alaffia’s AI agents. Our agentic systems are already in production reviewing real claims. As our customers’ needs grow more complex, we need someone to make our agents smarter, more accurate, and better grounded in the policies, contracts, and clinical evidence they reason over.

  • This is a model-focused role spanning four areas. You’ll design and build the agents themselves, including their planning, tool use, and multi-step reasoning across claims and medical records. You’ll develop and fine-tune the LLMs that power them on domain-specific healthcare data. You’ll build the RAG systems that keep their outputs accurate and traceable to source documents. And you’ll apply reinforcement learning techniques such as RLHF and DPO, turning feedback from our clinical auditors into measurable improvements in agent performance.

  • This role is for you if you like working at the frontier of applied LLM research and seeing your work land in production. You’ll take ideas from early experiments through to models that make real decisions on real claims, and you’ll have a direct hand in shaping how a Series B healthtech builds, trains, and improves its AI. You’ll work closely with our AI engineers, MLOps team, and clinical subject-matter experts.

Your Responsibilities

  • AI Agent Development: Design and build single- and multi-agent systems for claim review. This includes planning, tool use, memory, and multi-step reasoning across claims data, medical records, and coding guidelines.

  • RAG Development: Build and optimize retrieval-augmented generation pipelines over policies, contracts, clinical guidelines, and medical records. This covers chunking strategies, embeddings, hybrid search, re-ranking, and grounding and citation of outputs.

  • LLM Development & Fine-Tuning: Fine-tune open-source and proprietary LLMs (e.g., SFT, LoRA/QLoRA, instruction tuning) on domain-specific healthcare and claims data. Manage dataset curation, training runs, and model versioning.

  • Reinforcement Learning: Develop RL-based training and alignment methods (e.g., RLHF, DPO, reward modeling) that use human-in-the-loop feedback from clinical auditors. Use them to improve agent accuracy, reasoning, and decision quality.

  • Evaluation & Experimentation: Define evaluation frameworks, benchmarks, and metrics for agents, fine-tuned models, and RAG systems. These include hallucination, faithfulness, retrieval quality, and task-level accuracy. Run and track experiments rigorously.

  • Production Integration (in collaboration with the Software and DevOps teams): Take models and agents from prototype to production, optimizing for latency, cost, and reliability at scale.

  • Compliance & Safety: Ensure models and agents handle PHI securely and comply with HIPAA. Build guardrails that keep outputs accurate, explainable, and auditable.

What We’re Looking For

  • 3+ years of experience in applied ML or AI engineering, ideally with experience developing agentic systems, fine-tuning LLMs, and building RAG pipelines in production.

  • Strong software engineering fundamentals and proficiency in Python.

  • Highly experienced in developing AI agents and agentic workflows, including tool calling, orchestration, and multi-step reasoning

  • Hands-on experience with LLM fine-tuning (SFT, PEFT/LoRA) and training pipelines using frameworks such as PyTorch and Hugging Face

  • Proven experience building production RAG systems, including vector databases, embedding models, and retrieval optimization

  • Practical experience applying reinforcement learning to LLMs (e.g., RLHF, DPO, PPO, reward modeling)

  • Strong understanding of LLM evaluation methodologies and experiment tracking

Nice to have

  • Experience with agent frameworks such as LangGraph, LangChain, LlamaIndex, or similar

  • Familiarity with LLM provider APIs and self-hosted model serving (e.g., vLLM, TGI)

  • Experience with distributed training and GPU infrastructure

  • Working knowledge of AI tooling such as MLFlow, LangFuse, and Weights & Biases

  • Experience with healthcare data (claims, EHRs, medical coding such as ICD-10/CPT) or other regulated domains

Our Culture

Alaffia was born out of our founders’ personal connection to the inefficiency of the U.S. healthcare system. We are deeply mission-driven, with an abiding belief that technology can help create a better future for everyone — and we’re looking for others who share our passion for change to join the team. 

What Else Do You Get Working With Us?

  • Competitive Compensation

  • Medical, Dental, and Vision benefits

  • Flexible, paid vacation policy

  • Work in a flat organizational structure — direct access to Leadership

Skills Required

  • 3+ years of experience in applied machine learning or AI engineering
  • Strong software engineering fundamentals and proficiency in Python
  • Experience developing AI agents and agentic workflows, including tool calling, orchestration, and multi-step reasoning
  • Hands-on experience with LLM fine-tuning, including SFT and PEFT/LoRA
  • Experience using PyTorch and Hugging Face for training pipelines
  • Production experience building RAG systems, including vector databases, embedding models, and retrieval optimization
  • Practical experience applying reinforcement learning to LLMs, including RLHF, DPO, PPO, or reward modeling
  • Strong understanding of LLM evaluation methodologies and experiment tracking
  • Experience with agent frameworks such as LangGraph, LangChain, or LlamaIndex
  • Familiarity with LLM provider APIs and self-hosted model serving such as vLLM or TGI
  • Experience with distributed training and GPU infrastructure
  • Working knowledge of MLflow, Langfuse, or Weights & Biases
  • Experience with healthcare data, medical claims, EHRs, ICD-10, CPT, or other regulated domains

What the Team is Saying

Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
HQ: New York, NY
80 Employees
Year Founded: 2020

What We Do

Every year, U.S. health plans lose billions to improper payments and administrative waste. That wasted spending ultimately trickles down across the healthcare ecosystem, driving up costs for plans, providers, and patients alike. We’re here to change that paradigm. Alaffia is a new kind of claims operations partner for health plans. Using expert clinicians and transparent AI, we deliver deeper insights, smarter automation, and consistently better outcomes across the entire claims lifecycle. With Alaffia, health plans can cut wasted spending more effectively than ever and provide their members the most affordable care. We’re a high-growth, venture-backed Series B healthtech startup based in NYC and are actively scaling our company. Join us in helping to build a healthcare system that works better for everyone.

Gallery

Gallery
Gallery

Alaffia Health Offices

Hybrid Workspace

Employees engage in a combination of remote and on-site work.

We have a recognize the varying needs of our employees and have designed a location-based policy, offering flexibility while ensuring that the hybrid work model supports both individual and company success.

Typical time on-site: 3 days a week
HQNew York City (SoHo) HQ

Similar Jobs

Alaffia Health Logo Alaffia Health

Director of Account Management

Artificial Intelligence • Healthtech • Insurance • Machine Learning • Payments
Remote or Hybrid
United States
80 Employees
120K-160K Annually

Alaffia Health Logo Alaffia Health

Director, Payment Integrity Solutions

Artificial Intelligence • Healthtech • Insurance • Machine Learning • Payments
Remote or Hybrid
United States
80 Employees
150K-180K Annually

Alaffia Health Logo Alaffia Health

Clinical DRG Auditor

Artificial Intelligence • Healthtech • Insurance • Machine Learning • Payments
Remote or Hybrid
United States
80 Employees
85K-95K Annually

Alaffia Health Logo Alaffia Health

ML Ops Engineer

Artificial Intelligence • Healthtech • Insurance • Machine Learning • Payments
Hybrid
New York, NY, USA
80 Employees
145K-185K Annually

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account