[Remote] Staff AI Engineer

Note: The job is a remote job and is open to candidates in USA. Workato is a leader in enterprise infrastructure, focusing on unifying data, applications, processes, and AI into a single platform. They are seeking a Staff AI Engineer to lead the design and development of intelligent search systems, leveraging machine learning and advanced techniques in query understanding and ranking.

Responsibilities

• Lead the development of advanced query understanding systems that parse natural language, resolve ambiguity, and infer user intent
• Design and deploy learning-to-rank models that optimize relevance using behavioral signals, embeddings, and structured feedback
• Build and scale robust Entity Recognition pipelines that enhance document understanding, enable contextual disambiguation, and support entity-aware retrieval
• Architect next-gen search infrastructure capable of supporting highly dynamic document corpora and real-time indexing
• Create and maintain graph-based knowledge systems that enhance LLM capabilities through structured relationship data
• Drive improvements in query rewriting, intent classification, and semantic search, using both statistical and neural methods
• Own the design of evaluation frameworks for offline/online relevance testing, A/B experimentation, and continual model tuning
• Collaborate with product and applied research teams to translate user needs into data-informed search innovations
• Produce clean, scalable code and influence system architecture and roadmap across the relevance and platform stack

Skills

• Bachelors/Masters/PhD degree in Statistics, Mathematics or Computer Science, or another quantitative field
• 7+ years of backend engineering experience with 3+ years in search, information retrieval, or related fields
• Strong proficiency in Python
• Hands-on experience with search engines (Opensearch or Elasticsearch)
• Strong understanding of information retrieval concepts spanning traditional methods (TF-IDF, BM25) and modern neural search techniques (vector embeddings, transformer models)
• Experience with text processing, NLP, and relevance tuning
• Experience with relevance evaluation metrics (NDCG, MRR, MAP)
• Experience with large-scale distributed systems
• Strong analytical and problem-solving skills
• Proficiency in Knowledge Graph construction and optimization is a plus

Benefits

• Equity
• Benefits

Company Overview

Company H1B Sponsorship