Skip to content

Staff AI Engineer

175k – 220kUnited StatesML EngineeringRemote7+ YOE
Summary

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.

About the role

What You’ll Be Doing

  • Build and deliver AI solutions: Take ownership of developing high-performance AI features to help users detect, triage, and resolve incidents using observability data and tools.
  • Rapid experimentation and iteration: Implement a highly iterative process where you quickly prototype, test, and validate with real users, including shipping and evolving LLM- or agent-powered workflows for incident lifecycle management and automated analysis tasks.
  • Collaborate cross-functionally: Work with data analysts, product managers, and designers to shape AI-driven product features, including integration of agentic components with internal tools, alerting systems, runbooks, and developer workflows.
  • Utilize AI tools effectively: Use AI and automation tools to enhance both product functionality and your own development workflows.
  • Effective communication: Work in a highly dynamic and collaborative environment, communicating effectively and contributing across teams.
  • Ownership and impact: Take full ownership of the AI solutions you develop, ensuring they are innovative, scalable, maintainable, and aligned with real user workflows.

Requirements

  • Experience with LLMs, prompt engineering, and building applications powered by GenAI.
  • Proven track record of delivering software that made it into production and is actively used by users.
  • Exposure to working in cloud-native environments (e.g., AWS, GCP, Azure).
  • Experience using observability tools to understand and troubleshoot system behavior.
  • Strong engineering skills: Solid experience building production software systems (backend and/or full stack).
  • AI experience with a practical mindset: Familiar with AI technologies and frameworks, focused on delivering high-quality real-world solutions.
  • Quick iteration and experimentation: Comfortable releasing prototypes, collecting feedback, and iterating pragmatically.
  • Proven initiative: Take ownership, drive projects forward, deal with ambiguity, and define scope.
  • Collaborative attitude: Communicate effectively with peers, product managers, and designers; open to feedback with a solutions-oriented mindset.

Bonus Points

  • Experience building or working with agent frameworks or multi-agent workflows.
  • Experience with infrastructure/devops tooling: Kubernetes, Docker, Terraform or similar.
  • Familiarity with model fine-tuning techniques.
  • Experience building observability tooling.

Compensation & Benefits

  • Base compensation range: USD 174,986 - USD 220,000.
  • Restricted Stock Units (RSUs) included.
  • Bonus (if applicable) and other benefits.
Skills
LLMsPrompt EngineeringGenAIBackend DevelopmentFull Stack DevelopmentAWSGCPAzureObservability ToolsAgent FrameworksKubernetesDockerTerraformModel Fine-Tuning
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++
Talkiatry

Senior AI Engineer

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.

170k – 195kUnited StatesML EngineeringRemote2+ YOERAGReact
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