Talent.com
AI / ML Engineer (LLM Optimization & AI-Driven Workflows)

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

BiomichealthSan Francisco, CA, United States
21 days ago
Job type
  • Full-time
Job description

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.

Overview

Why This Role?

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.

Responsibilities

  • 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.
  • 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.
  • Qualifications

  • 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).
  • Benefits

  • $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.
  • #J-18808-Ljbffr

    Create a job alert for this search

    Optimization Engineer • San Francisco, CA, United States

    Related jobs
    • Promoted
    • New!
    Engineer - Machine Learning & AI

    Engineer - Machine Learning & AI

    DishFoster City, CA, United States
    Full-time
    EchoStar is reimagining the future of connectivity.Our business reach spans satellite television service, live-streaming and on-demand programming, smart home installation services, mobile plans an...Show moreLast updated: 18 hours ago
    • Promoted
    ML Engineer

    ML Engineer

    PhizenixMenlo Park, CA, US
    Full-time +1
    Client Opportunity | Through Phizenix.Phizenix, a certified minority and women-led recruiting firm, is hiring on behalf of an innovative generative AI startup that's developing diffusion-based ...Show moreLast updated: 30+ days ago
    • Promoted
    AIML - ML Research Engineer, UI Understanding

    AIML - ML Research Engineer, UI Understanding

    Apple Inc.San Francisco, CA, United States
    Full-time
    San Francisco, California, United States Machine Learning and AI.We are building machine learning technologies to understand, interact with, evaluate and generate user interfaces.These technologies...Show moreLast updated: 30+ days ago
    • Promoted
    Machine Learning Engineer - GenAI, LLM, Agentic AI

    Machine Learning Engineer - GenAI, LLM, Agentic AI

    CerebrasSanta Clara, CA, United States
    Full-time
    We are building the next generation of our AI-powered talent platform, aiming to match the right career for everyone in the world. Our AI-native enterprise talent intelligence platform leverages Gen...Show moreLast updated: 20 days ago
    • Promoted
    Senior Applied AI Engineer – ML for Systems & Infrastructure

    Senior Applied AI Engineer – ML for Systems & Infrastructure

    Databricks Inc.San Francisco, CA, United States
    Full-time
    Senior Applied AI Engineer – ML for Systems & Infrastructure.The Applied AI team at Databricks sits at the forefront of advancing GenAI-powered products. Over the past years, we’ve launched Databric...Show moreLast updated: 18 days ago
    • Promoted
    AI / ML Engineer

    AI / ML Engineer

    FoodHealth CompanySan Francisco, California, United States
    Full-time
    We are on a mission to improve the world's health through food.Through our flagship product, the FoodHealth Score, we provide data tools that empower consumers to make better choices, help retailer...Show moreLast updated: 1 day ago
    AIML - ML Research Engineer, UI Understanding

    AIML - ML Research Engineer, UI Understanding

    MediabistroSan Francisco, CA, United States
    Full-time
    San Francisco, California, United States Machine Learning and AI.We are building machine learning technologies to understand, interact with, evaluate and generate user interfaces.These technologies...Show moreLast updated: 30+ days ago
    • Promoted
    Machine Learning Engineer, GenAI & LLM - AiDP - IS&T

    Machine Learning Engineer, GenAI & LLM - AiDP - IS&T

    Apple Inc.Sunnyvale, CA, United States
    Full-time
    Machine Learning Engineer, GenAI & LLM - AiDP - IS&T.Sunnyvale, California, United States Corporate Functions.As a pivotal member of Apple’s enterprise generative AI efforts, you will : - Innovate tr...Show moreLast updated: 30+ days ago
    AIML - ML Research Engineer, UI Understanding Job at Apple Inc. in San Francisco

    AIML - ML Research Engineer, UI Understanding Job at Apple Inc. in San Francisco

    MediabistroSan Francisco, CA, United States
    Full-time
    San Francisco, California, United States Machine Learning and AI.We are building machine learning technologies to understand, interact with, evaluate and generate user interfaces.These technologies...Show moreLast updated: 30+ days ago
    • Promoted
    Engineer II - Generative AI & ML

    Engineer II - Generative AI & ML

    DishFoster City, CA, United States
    Full-time
    EchoStar is reimagining the future of connectivity.Our business reach spans satellite television service, live-streaming and on-demand programming, smart home installation services, mobile plans an...Show moreLast updated: 30+ days ago
    • Promoted
    AI / ML Engineer

    AI / ML Engineer

    General MotorsSan Francisco, CA, United States
    Full-time
    As an AI / ML Engineer on the Metrics Frameworks team, part of the Simulation, Evaluation, and Data organization, you will be an individual contributor focused on developing and optimizing infrastruc...Show moreLast updated: 1 day ago
    • Promoted
    Sr Machine Learning Engineer - GenAI, LLM, Agentic AI

    Sr Machine Learning Engineer - GenAI, LLM, Agentic AI

    CerebrasSanta Clara, CA, United States
    Full-time
    We are building the next generation of our AI-powered talent platform, aiming to match the right career for everyone in the world. Our AI-native enterprise talent intelligence platform leverages Gen...Show moreLast updated: 20 days ago
    • Promoted
    ML / AI Engineer

    ML / AI Engineer

    RilletSan Francisco, CA, United States
    Full-time
    Our customers are the financial brains of their companies.Our job is to help them run the numbers with impossible speed, accuracy, and insight. Today, we do that with powerful and elegant accounting...Show moreLast updated: 30+ days ago
    • Promoted
    AI / ML Engineer

    AI / ML Engineer

    PRAGMATIKESan Francisco, CA, US
    Full-time
    On-site Cambridge, MA (Eastern Time / UTC -4).This role is on-site in in the.Please inquire for more details during the interview process. We are hiring at Pragmatike to expand our team and drive th...Show moreLast updated: 30+ days ago
    • Promoted
    ML Research Engineer, ML Systems

    ML Research Engineer, ML Systems

    Scale AI, Inc.San Francisco, CA, United States
    Full-time
    Scale's ML platform (RLXF) team builds our internal distributed framework for large language model training and inference. The platform has been powering MLEs, researchers, data scientists and opera...Show moreLast updated: 30+ days ago
    AI / ML Engineer

    AI / ML Engineer

    Cloudious LLCMountain View, CA, United States
    Full-time
    Quick Apply
    Help build GenAI solutions from prototype to production.Lead prompt engineering : system / tool prompts, function calling, prompt versioning with offline / online evals.Implement eval...Show moreLast updated: 1 day ago
    • Promoted
    Senior AI / ML engineer

    Senior AI / ML engineer

    Storm3Fremont, CA, United States
    Full-time
    Senior Data / AI / ML Engineer.Series A HealthTech | Clinical Data Intelligence Platform.San Francisco (Hybrid | US Only). A Series A HealthTech company is transforming how clinicians use healthcare...Show moreLast updated: 1 day ago
    • Promoted
    Sr. AI / ML Engineer

    Sr. AI / ML Engineer

    VectraSan Jose, CA, United States
    Full-time
    Vectra is the leader in AI-driven threat detection and response for hybrid and multi-cloud enterprises.The Vectra AI Platform delivers integrated signal across public cloud, SaaS, identity, and dat...Show moreLast updated: 11 days ago