Staff AI/ML Engineer, MyHealthTeam

Swoop Airlines and Aviation

1d • Remote

About The Position

MyHealthTeam builds communities for people living with chronic and rare conditions. We reach millions of people each month, and we’re investing deeply in AI to help members find the right support, content, and recommendations — safely, responsibly, and at scale. We’re looking for a Staff AI/ML Engineer who is product-minded, hands-on, and excited to ship production ML/LLM systems that turn messy, real-world text and behavioral data into reliable experiences for patients.

Requirements

• 8+ years building and shipping production ML systems (or equivalent experience with demonstrable impact)
• Strong Python skills and experience with ML/LLM libraries and tooling (e.g., Hugging Face ecosystem, LangChain/LangGraph, or equivalent)
• Proven ability to design production-grade pipelines (training/inference/eval) and operate models in real systems (monitoring, rollbacks, incident handling)
• Solid grounding in ML fundamentals (NLP, deep learning, statistical reasoning, evaluation)
• Experience with MLOps best practices: versioning, reproducibility, CI/CD, model registry patterns, feature/data management, and infrastructure collaboration
• Experience working with large-scale data using Databricks/Spark or equivalent distributed processing
• Strong product and stakeholder instincts: you can translate ambiguous business needs into measurable ML outcomes

Nice To Haves

• Experience building RAG and retrieval systems: vector databases, hybrid search, ranking, recommendation, query understanding
• Experience in healthcare or regulated environments, including privacy-by-design, auditability, and safety reviews (HIPAA/PHI familiarity a plus)
• Experience with streaming/clickstream data, experimentation platforms, and causal/measurement thinking
• Ability to prototype end-to-end experiences (e.g., Streamlit, Gradio, lightweight frontends)
• Experience designing LLM safety systems: red-teaming, adversarial testing, prompt injection mitigation, output filtering, human-in-the-loop review

Responsibilities

• Build end-to-end ML/LLM features from problem definition → data → modeling → evaluation → deployment → monitoring.
• Develop LLM applications with retrieval and tool use (e.g., RAG, orchestration/workflows, structured extraction) to deliver trustworthy consumer health experiences.
• Convert unstructured text (posts, comments, messages, search queries) into structured signals (topics, entities, intent, sentiment, safety flags) using a mix of classical NLP and modern LLMs.
• Create and maintain data pipelines for training, inference, evaluation, and analytics (batch and/or streaming as needed).
• Design evaluation systems that measure quality and safety: offline metrics, golden datasets, human review workflows, and online A/B testing alignment.
• Implement production guardrails to reduce harm and misinformation risk (policy constraints, refusal behavior, citations/attribution when appropriate, red-teaming, monitoring, and incident response).
• Set up monitoring for model + system health (latency, cost, drift, regressions, quality metrics).
• Partner closely with the Product, Engineering, and Data teams and clinical/subject-matter experts to validate outputs and define what “correct” means for sensitive, health-adjacent use cases.
• (Staff scope) Lead architecture and technical direction for applied AI across the organization; mentor engineers; establish best practices and reusable platforms.

Benefits

• Mission-driven work with massive reach: help millions of people find support and better health outcomes
• High-ownership culture: small teams, fast shipping, visible impact
• Strong collaboration: product, data, and domain experts working together
• A chance to shape applied AI in a real consumer product with real constraints

What This Job Offers

• Job Type: Full-time
• Career Level: Senior
• Education Level: No Education Listed