Job Details

Join to apply for the AI/ML Engineer (LLM Optimization & AI-Driven Workflows) role at Legion Health

Join to apply for the AI/ML Engineer (LLM Optimization & AI-Driven Workflows) role at Legion Health

This range is provided by Legion Health. Your actual pay will be based on your skills and experience — talk with your recruiter to learn more.

Base pay range

$120,000.00/yr - $180,000.00/yr

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

faster, smoother, and more accessible.

We're hiring

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.

Seniority level

• Seniority levelMid-Senior level

Employment type

• Employment typeFull-time

Job function

• Job functionEngineering and Information Technology
• IndustriesMental Health Care

Referrals increase your chances of interviewing at Legion Health by 2x

Sign in to set job alerts for “Machine Learning Engineer” roles.

San Francisco, CA $130,000.00-$230,000.00 5 months ago

Staff Software Engineer, AI/ML Recommendations, Rankings, Predictions, YouTube

San Francisco, CA $140,000.00-$180,000.00 5 months ago

Machine Learning Engineer (I, II, or Sr.)

San Francisco, CA $140,000.00-$160,000.00 5 months ago

Research Engineer - Machine Learning (ML)

San Francisco, CA $140,000.00-$160,000.00 4 months ago

San Francisco, CA $130,000.00-$238,000.00 2 weeks ago

San Francisco, CA $100,000.00-$180,000.00 1 year ago

San Francisco, CA $150,000.00-$250,000.00 3 weeks ago

San Francisco, CA $150,000.00-$260,000.00 4 months ago

San Mateo, CA $140,000.00-$210,000.00 3 weeks ago

San Francisco, CA $140,000.00-$290,000.00 7 months ago

Machine Learning Engineer, GenAI Applied ML

San Francisco, CA $176,000.00-$220,000.00 3 weeks ago

San Francisco, CA $160,000.00-$180,000.00 2 days ago

San Francisco, CA $100,000.00-$300,000.00 2 weeks ago

Machine Learning Engineers (Open-Endedness) - Open Level

San Francisco, CA $175,000.00-$225,000.00 7 months ago

San Francisco, CA $200,000.00-$300,000.00 4 months ago

Machine Learning Engineer, Identity Product

San Francisco, CA $140,000.00-$200,000.00 4 days ago

San Mateo, CA $195,000.00-$255,000.00 7 months ago

San Francisco, CA $180,000.00-$270,000.00 3 months ago

San Bruno, CA $119,000.00-$169,000.00 1 month ago

We're unlocking community knowledge in a new way. Experts add insights directly into each article, started with the help of AI.

#J-18808-Ljbffr