Soma Capital Portfolio Jobs

AI/ML Engineer đź§  (LLM Optimization & AI-Driven Workflows)

Legion Health

Legion Health

Software Engineering, Data Science
United States
USD 130k-180k / year + Equity
Posted on Feb 11, 2025
🚀 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.