Sr. Staff AI Engineer (Remote)
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 job
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
Python, AWS, GCP, JavaScript, TypeScript, Java, Go, LLMs, RAG, LangChain, Llamaindex, Dspy, Pinecone, Weaviate, MLOps
Similar jobs
ML Engineering jobs
Harvey
San Francisco, CA
Leads the design and operation of reliable, scalable model infrastructure powering AI inference across multiple providers. Requires 7+ years of distributed-systems engineering experience, strong programming skills, and expertise in production reliability and cloud infrastructure.
Shield AI
San Mateo, CA
Leads the design, implementation, integration, and field validation of tactical autonomy and multi-agent coordination capabilities for unmanned platforms. Requires 7+ years of relevant experience, production C++, technical leadership, and eligibility for a U.S. Secret clearance.
Snowflake
Bellevue, WA
Leads the roadmap and technical vision for Snowflake Feature Store, building reliable, high-performance machine learning platform capabilities and supporting technical execution across partner teams. Requires 10+ years of experience with data-serving infrastructure or ML platforms, plus Java and Python expertise.
Shield AI
Washington, DCLeads development and integration of advanced maritime autonomy for USVs, UUVs, and cooperating UAVs, including motion planning, localization, safety, and multi-agent coordination. Requires staff-level technical leadership, substantial robotics experience, C++ and Python proficiency, and eligibility for a SECRET clearance.
Ironclad
San Francisco, CA
Leads architecture and technical direction for agentic search systems combining LLMs, retrieval, and content-understanding pipelines for contract intelligence. The role requires 10+ years building production systems, deep search or LLM expertise, and strong cross-team technical leadership.