Skip to content

Applied AI - Senior/ Staff Software Engineer, Lead

225k – 260kSan Francisco, CAMenlo Park, CAML EngineeringHybrid
Summary

Leads development of production AI systems using LLMs, RAG, and multi-agent architectures to enhance clinical workflows, patient risk prediction, and care coordination in healthcare. Ships end-to-end features with cross-functional collaboration in a fast-paced startup.

About the role

Responsibilities

  • Build AI-powered features that improve patient care, coordination, and clinical workflows
  • Develop production systems using LLMs, RAG, embeddings, and multi-agent architectures
  • Ship products like clinical visit summaries, patient risk models, and voice coordination assistants
  • Design workflow agents that automate scheduling, follow-ups, and operational tasks
  • Collaborate with product, clinical ops, and data partners to deliver measurable results
  • Help define the technical direction, tooling, and standards for a brand-new team

Requirements

  • Built and shipped production software end-to-end in fast-moving environments
  • Worked with AI/ML systems such as LLMs, RAG pipelines, embeddings, or applied NLP
  • Navigated ambiguity and iterated quickly with tight feedback loops
  • Designed systems with an eye toward scalability, tradeoffs, and real-world constraints
  • Collaborated with cross-functional teams to deliver impact

(Bonus) Worked with healthcare data formats like FHIR, HL7, or clinical documentation

What gives you an edge

  • You've shipped AI products—not just prototypes or research demos
  • You've built systems that interface with real users or complex workflows
  • You've operated in early-stage or 0→1 environments with high ownership
  • You understand reliability, iteration speed, and production safety
  • You're motivated by solving real problems for real people, not AI hype

Tech stack

AI & Modeling
LLMs, RAG pipelines, embeddings
LangChain / LlamaIndex or similar
Python

Backend & Services
Node.js / TypeScript
GraphQL / REST APIs
AWS Lambda, AppSync, DynamoDB

Data & Infrastructure
BigQuery, Elasticsearch / OpenSearch
Airflow or similar orchestration
Vector stores and retrieval systems

Bonus Tools & Domains
FHIR, HL7, clinical notes
Voice agents / telephony APIs
Statsig or internal experimentation tooling

Skills
LLMsRAGPythonNode.jsTypeScriptGraphQLAWS LambdaDynamoDBBigQueryLangChainLlamaIndexFHIRHL7AirflowElasticsearch
Similar roles at this salary range
All ML Engineering jobs →
Airbnb

Staff Machine Learning Engineer

Build and deploy cutting-edge ML and Generative AI systems to transform Airbnb's customer support experience, focusing on LLM fine-tuning, RAG, and intelligent service automation.

212k – 260kSan Francisco, CAML EngineeringRemote9+ YOELLMRAG
Mixpanel

Senior Software Engineer, AI Platform

Senior Software Engineer building scalable AI infrastructure, agent orchestration frameworks, evaluation systems, and high-performance LLM serving at Mixpanel. Requires 5+ years experience and hands-on LLM/agent work.

226k – 306kSan Francisco, CAML EngineeringHybrid5+ YOELLMsMLOps
Twilio

Tech Lead, Applied Research

Tech Lead driving AI R&D and end-to-end delivery of production-ready prototypes using full-stack development, LLMs, and emerging technologies. Requires 10+ years experience and strong autonomy.

228k – 335kUnited StatesML EngineeringRemote10+ YOEGoSQL
Reddit

Senior Machine Learning Systems Engineer

Build large-scale ML experimentation and training orchestration platforms, including agentic AI execution systems, to accelerate Ads ML development at Reddit. Requires 5+ years infrastructure experience and 2+ years building production ML platforms.

217k – 303kUnited StatesML EngineeringRemote5+ YOERayArgo
Axion

Staff Software Engineer, Agentic Platform

Senior individual contributor architecting and scaling agentic LLM systems that turn messy manufacturing data into reliable root-cause insights. Owns orchestration, retrieval, evaluation, and guardrails for non-deterministic production systems.

250k – 270kSan Francisco, CA +1ML EngineeringHybrid7+ YOEMCPobservability