Skip to content

Senior Software Engineer, AI Platform

173k – 251kOakland, CAML EngineeringHybrid4+ YOE
Summary

Builds and operates AI platform capabilities including RAG pipelines, semantic retrieval, agentic orchestration, and LLM integrations to power legal tech products. Requires 4+ years in distributed cloud systems, AI/ML experience, and proficiency in modern programming languages.

About the role

Responsibilities

  • Build AI platform capabilities powering product experiences like Deep Dive, predictive coding, multi-modal understanding, agentic workflows, translations, search, and review.
  • Contribute to RAG, semantic retrieval, and agentic orchestration patterns, including indexing pipelines, query flows, tool-calling, planning logic, relevance tuning, benchmarking, and multi-modal workloads.
  • Collaborate with Product, Platform, Security, and DevOps to build and ship features in production.
  • Scale system to handle hundreds of millions of documents.
  • Provide technical mentorship to engineers and become expert in codebase.
  • Perform code reviews using knowledge of languages, design patterns, and best practices.
  • Fix defects by triaging issues and writing code to debug, track, and resolve.
  • Provide on-call support, answering customer questions, triaging defects, and carrying pager.
  • Contribute to internal and external documentation.
  • Conduct technical interviews for recruiting.

Requirements

  • BS or MS in Computer Science or equivalent.
  • Proficiency in coding with C, C++, C#, Java, Python, JavaScript, Go, or Rust.
  • Knowledge of algorithms, fundamental CS concepts, relational databases, API design, and building user interfaces.
  • Practical experience with AI/ML systems: retrieval pipelines, semantic search, agentic pipelines, document classification, or LLM-integrated features.
  • Experience with AI tools like Cursor and Claude Code.
  • At least 4 years building distributed cloud systems with service-based architecture, frontend frameworks for rich web apps, and best practices for testing, maintaining, launching cloud software.
  • At least 1 year leading or coordinating multi-developer efforts.

Nice-to-haves

  • Familiarity with AWS.
  • Familiarity with MySQL, Postgres.
  • Familiarity with vector or graph databases.
  • Familiarity with LLMOps.

Compensation

Salary range: $173,000 - $251,000 (dependent on experience and skills).
Equity program, 401(k) matching, health/dental/vision, FSA, paid parental leave, sick leave, vacation, holidays, mental health benefits, L&D stipend, life/disability insurance.

Skills
PythonJavaScriptJavaGoRustRAGLLMsemantic searchvector databasesAWSPostgresMySQLdistributed systemscloud architecturefrontend frameworks
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
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