Staff AI Engineer jobs in United States
cer-icon
Apply on Employer Site
company-logo

Workato · 16 hours ago

Staff AI Engineer

Workato is a leader in enterprise orchestration, helping businesses streamline operations through its AI-powered platform. The Staff AI Engineer will lead the design, development, and optimization of intelligent search systems, leveraging machine learning to enhance search relevance and user experience.

Cloud ComputingData IntegrationMarketing AutomationSaaSSales Automation
check
H1B Sponsor Likelynote

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

Qualification

Search RelevanceMachine LearningInformation RetrievalPythonSearch EnginesNLPRelevance Evaluation MetricsDistributed SystemsKnowledge GraphAnalytical SkillsCommunication SkillsCollaborationDetail-orientedSelf-motivatedProblem-solving

Required

Bachelor's/Master's/PhD degree in Statistics, Mathematics, 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
Strong communication abilities to explain technical concepts
Collaborative mindset for cross-functional teamwork
Detail-oriented with strong focus on quality
Self-motivated and able to work independently
Passion for solving complex search problems

Preferred

Proficiency in Knowledge Graph construction and optimization is a plus

Company

Workato is an automation and orchestration platform designed to connect applications, data sources, and workflows across an organization.

H1B Sponsorship

Workato has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (10)
2024 (2)
2023 (7)
2022 (12)
2021 (6)
2020 (6)

Funding

Current Stage
Late Stage
Total Funding
$415M
Key Investors
Digital Industry SingaporeBattery VenturesRedpoint
2025-11-07Grant
2021-11-10Series E· $200M
2021-01-12Series D· $110M

Leadership Team

leader-logo
Vijay Tella
Co-founder and CEO
linkedin
leader-logo
Adam Seligman
Chief Technology Officer and GM
linkedin
Company data provided by crunchbase