Skip to content

Senior AI Engineer

170k – 195kUnited StatesML EngineeringRemote2+ YOE
Summary

Build full-stack AI systems including agentic workflows, RAG pipelines, and production infrastructure for mental healthcare applications. Requires 2+ years software engineering experience and 1+ year with LLMs or agentic AI.

About the role

Responsibilities

  • Build frontend interfaces, backend services, APIs, integrations, and data flows that make AI capabilities usable in patient, clinician, and internal workflows.
  • Build and iterate on agentic systems that use reasoning, tool use, retrieval, memory, and multi-step workflows to solve healthcare problems.
  • Build and maintain RAG pipelines, ingestion workflows, retrieval tuning, semantic search, and knowledge quality improvements.
  • Implement evaluation frameworks, automated tests, experiments, regression checks, and quality metrics to improve AI system performance over time.
  • Implement guardrails, fallback flows, human-in-the-loop workflows, access controls, tracing, monitoring, and diagnostics for production AI systems.
  • Contribute to reliable production rollouts, CI/CD workflows, debugging, incident response, performance improvements, and operational efficiency.
  • Work with senior engineers, product, clinical, design, and data partners to translate real workflows into safe, reliable AI-powered experiences.

Requirements

  • 2+ years of software engineering experience, including building, testing, and maintaining production software systems.
  • 1+ year of experience building with LLMs, agentic AI, or production AI systems, ideally beyond simple chatbot or prompt-only use cases.
  • Strong programming skills in Python and/or TypeScript, with experience building backend services, APIs, integrations, data flows, or full-stack product features.
  • Experience with modern frontend or full-stack development using technologies such as React, Next.js, TypeScript, Node.js, or similar frameworks.
  • Experience contributing to AI systems involving retrieval, tool use, workflow orchestration, structured outputs, context management, or multi-step execution.
  • Familiarity with agentic frameworks and orchestration tools such as LangGraph, CrewAI, OpenAI Agents SDK, or similar frameworks.
  • Familiarity with RAG systems, vector databases, semantic search, retrieval infrastructure, model serving, or knowledge pipelines.
  • Familiarity with cloud-native development, containerized services, Kubernetes, CI/CD workflows, or scalable production environments.
  • Strong debugging, communication, and collaboration skills, with the ability to work through ambiguity and make steady progress with guidance from senior engineers.

Nice to Have

  • Experience building AI systems in healthcare, clinical, safety-critical, or other regulated environments.
  • Familiarity with HIPAA, clinical AI safety, FDA AI/ML guidance, or compliance-aware software development.
  • Experience using tests, evaluation methods, quality metrics, experiments, observability, or release checks to improve system reliability and performance.
Skills
PythonTypeScriptReactNext.jsNode.jsLangGraphCrewAIOpenAI Agents SDKRAGKubernetes
Similar roles at this salary range
All ML Engineering jobs →
Zoox

Machine Learning Engineer - Simulation Framework

Machine Learning Engineer focused on GPU-based simulation frameworks, reinforcement learning, and bridging sim-to-real gaps for autonomous vehicle safety validation. Requires MS/PhD and strong C++/Python experience.

151k – 257kFoster City, CA +1ML EngineeringHybrid7+ YOEJAXC++
Grafana Labs

Staff AI Engineer

Staff AI Engineer building and shipping LLM/agent-powered observability features for incident detection, triage, and resolution. Requires strong production software engineering experience plus practical GenAI/LLM application skills.

175k – 220kUnited StatesML EngineeringRemote7+ YOEAWSGCP
Pinterest

Staff Software Engineer, Trends Machine Learning Infrastructure

Lead technical direction for Pinterest's unified AI-powered Trends and Audience Insights platform. Architect scalable ML data pipelines and LLM capabilities while mentoring engineers and driving cross-team integrations.

177k – 365kSan Francisco, CAML EngineeringHybrid8+ YOELLMsCodex
Airbnb

Machine Learning Engineer

Build and deploy cutting-edge Agentic AI and LLM systems to transform Airbnb's customer service experience. Requires PhD or equivalent experience and production ML/AI deployment expertise.

170k – 180kSan Francisco, CA +1ML EngineeringOn-site3+ YOELLMSFT
Parallel

Member of Technical Staff, Model Training

Own the training pipeline for search and agent models, building from product usage data through fine-tuning and evaluation to production deployment. Requires deep expertise in transformer fine-tuning, data curation, and training models for ranking, retrieval, and agent behavior.

150k – 300kCaliforniaML EngineeringOn-siteData curationLabel quality