Legion Health

World-class, AI-enabled psychiatry covered by insurance

AI/ML Engineer 🧠 (LLM Optimization & AI-Driven Workflows)

$130K - $180K / 0.40% - 1.20%
Location
Austin, TX, US / San Francisco, CA, US / Remote (US)
Job Type
Full-time
Experience
3+ years
Connect directly with founders of the best YC-funded startups.
Apply to role ›
Daniel Wilson
Daniel Wilson
Founder

About the role

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

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 + 0.4-1.2% 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.

About Legion Health

Legion Health is a pioneering telepsychiatry company dedicated to transforming patient care through advanced technology. We leverage cutting-edge language models and robust data analytics to enhance our patient-provider interactions, improve access to mental health services, and streamline our clinical operations.

Our mission is to integrate innovative AI solutions into every facet of psychiatric care, ensuring providers and patients benefit from the most effective and efficient treatment strategies.

Legion Health
Founded:2021
Team Size:9
Status:
Active
Location:Austin, TX
Founders
Arthur MacWaters
Arthur MacWaters
Founder
Yash Patel
Yash Patel
CEO
Daniel Wilson
Daniel Wilson
Founder