Job description

To be considered, you must apply directly at the following link: https://www.ycombinator.com/companies/legion-health/jobs/26GxO6f-ai-ml-engineer-llm-optimization-ai-driven-workflows

🚀 Legion Health | AI-Driven Psychiatric Care – We’re Hiring!

To be considered, you must apply directly at the following link: https://www.ycombinator.com/companies/legion-health/jobs/26GxO6f-ai-ml-engineer-llm-optimization-ai-driven-workflows

Please make sure to submit your application through this link—this is the only way we’re reviewing candidates for this position.

Join us in building the most efficient, AI-powered mental healthcare system.

We’re a YC-backed company revolutionizing telepsychiatry—not with AI for diagnostics, but with AI-driven operations that actually make patient care faster, smoother, and more accessible.

We’re hiring top-tier engineers who thrive in high-velocity, AI-first environments and want to build technology that meaningfully improves lives. If you love shipping fast, working with AI, and solving hard problems, keep reading. 👇

Why This Role?

You’ll work on practical AI deployment—optimizing LLM-powered workflows for real-world mental healthcare.
You’ll refine AI models that improve scheduling, risk assessment, and revenue cycle automation.
You’ll optimize feedback loops to improve LLM-driven decisions over time.
You’ll shape a real-world reinforcement learning approach that trains AI on provider feedback.

What You’ll Work On

Optimizing LLM workflows (fine-tuning models, prompt engineering, integrating feedback loops).
Improving AI explainability—building systems that show providers why AI makes certain recommendations.
Implementing retrieval-augmented generation (RAG) for more accurate, context-aware AI responses.
Building tooling to measure AI override rates, decision accuracy, and improvement metrics.
Working closely with clinicians to ensure AI improves their efficiency—not replaces them.

What We’re Looking For

3+ years experience in AI/ML engineering.
Strong Python & ML background (LLMs, NLP, PyTorch/TensorFlow).
Experience with reinforcement learning (or curiosity to implement RL-based workflows).
Comfortable with production AI deployment (vector databases, API integrations, etc.).
Interest in AI for healthcare (compliance/security mindset preferred).

What We Offer

$130-180K salary + equity.
Real AI autonomy—you won’t just be fine-tuning prompts, you’ll be shaping LLM-driven patient care workflows.
Work at the bleeding edge of AI deployment in healthcare.
A mission with impact—we’re solving real problems that affect millions of patients.

Please make sure to submit your application through this link—this is the only way we’re reviewing candidates for this position.

Required profile