[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.ResponsibilitiesLead the development of advanced query understanding systems that parse natural language, resolve ambiguity, and infer user intentDesign and deploy learning-to-rank models that optimize relevance using behavioral signals, embeddings, and structured feedbackBuild and scale robust Entity Recognition pipelines that enhance document understanding, enable contextual disambiguation, and support entity-aware retrievalArchitect next-gen search infrastructure capable of supporting highly dynamic document corpora and real-time indexingCreate and maintain graph-based knowledge systems that enhance LLM capabilities through structured relationship dataDrive improvements in query rewriting, intent classification, and semantic search, using both statistical and neural methodsOwn the design of evaluation frameworks for offline/online relevance testing, A/B experimentation, and continual model tuningCollaborate with product and applied research teams to translate user needs into data-informed search innovationsProduce clean, scalable code and influence system architecture and roadmap across the relevance and platform stackSkillsBachelors/Masters/PhD degree in Statistics, Mathematics or Computer Science, or another quantitative field7+ years of backend engineering experience with 3+ years in search, information retrieval, or related fieldsStrong proficiency in PythonHands-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 tuningExperience with relevance evaluation metrics (NDCG, MRR, MAP)Experience with large-scale distributed systemsStrong analytical and problem-solving skillsProficiency in Knowledge Graph construction and optimization is a plusBenefitsEquityBenefitsCompany OverviewWorkato is an automation and orchestration platform designed to connect applications, data sources, and workflows across an organization. It was founded in 2013, and is headquartered in Mountain View, California, USA, with a workforce of 1001-5000 employees. Its website is http://www.workato.com.Company H1B SponsorshipWorkato has a track record of offering H1B sponsorships, with 10 in 2025, 2 in 2024, 7 in 2023, 12 in 2022, 6 in 2021, 6 in 2020. Please note that this does not guarantee sponsorship for this specific role.