Skip to content

Sr. Staff AI Engineer (Remote)

229k – 284kLos Angeles, CACaliforniaRemote10+ YOE
Summary

Builds production-grade generative AI systems powering patient engagement, provider workflows, and clinical operations in mental healthcare. Requires 10+ years software engineering, deep Python/cloud expertise, and 2+ years scaling AI products with foundation models and LLM patterns.

About the role

Required Qualifications

  • 10+ years of software engineering, including 7+ years of experience with Public Cloud (AWS, GCP, etc.), 5+ years designing and scaling distributed systems.
  • 5+ years of deep experience in Python and at least one backend language (preferably JavaScript, TypeScript, Java, or Go)
  • 2+ years of proven track record building AI-powered products at scale, including hands-on experience with foundation models (OpenAI, Anthropic, Bedrock, Gemini, etc.).
  • 2+ years of strong experience in LLM integration patterns (RAG, function calling, agents, memory routing, etc.), or AI orchestration frameworks (eg. LangChain, LlamaIndex, DSPy, Haystack), or vector databases (eg. Pinecone, Weaviate, FAISS, Milvus).
  • 3+ years in working with MLOps, data pipelines, evaluation, and observability systems for continuous model improvement.

Preferred Qualifications

  • Experience architecting secure and compliant AI solutions in regulated environments (HIPAA, GDPR, etc.).
  • Familiarity with human-in-the-loop systems and clinical decision support frameworks.
  • Experience designing evaluation pipelines for human alignment, factual accuracy, or model interpretability.
  • Contributions to open-source AI frameworks or applied research in NLP, healthcare AI, or GenAI safety.
  • Experience contributing to early-stage team growth (0 → 1)
  • Experience leading or mentoring engineering teams in AI, ML platform, or applied research domains.
Skills
PythonAWSGCPJavaScriptTypeScriptJavaGoLLMRAGLangChainLlamaIndexDSPyPineconeWeaviateMLOps
Similar roles at this salary range
All ML Engineering jobs →
Coinbase

Staff Machine Learning Engineer

Staff ML Engineer leading end-to-end identity verification ML systems including document authenticity, face matching, liveness detection, GNN-based identity graphs, and behavioral risk models. Requires 8+ years production ML experience and domain expertise in biometrics or fraud detection.

218k – 257kUnited StatesML EngineeringRemote8+ YOENLPLLMs
Hinge Health

Staff Machine Learning Scientist

Own ML systems for send-time optimization, propensity modeling, and nudge decisions at consumer scale. Set experimentation standards and mentor a small ML team.

205k – 307kSan Francisco, CAML EngineeringHybrid7+ YOESQLdbt
Docker

Staff ML Engineer

Founding Staff ML Engineer building production ML systems for governance, security, and agentic platform capabilities at Docker. Owns architecture, data pipelines, evaluation, and model lifecycle while mentoring the growing team.

205k – 330kPalo Alto, CA +1ML EngineeringRemote8+ YOELLMsRetrieval
Pinterest

Principal Engineer, AI Platform

Principal Engineer setting technical vision and building AI/ML infrastructure for Generative AI and Recommender Systems at Pinterest, scaling to hundreds of millions of inferences per second. Requires deep expertise in distributed systems and proven cross-org technical leadership.

243k – 500kSan Francisco, CAML EngineeringHybrid7+ YOEC++Java
Reddit

Senior Research Engineer, Post-training & Evaluation

Own evaluation science and post-training methodology for Reddit's foundational LLMs. Define benchmarks, design model-as-a-judge systems, and set SFT recipes that turn base models into safe, Reddit-native endpoints.

230k – 322kUnited StatesML EngineeringRemote6+ YOESFTCPT