← All roles

Staff Software Engineer - Machine Learning (Search)

Databricks· Engineering
Apply on company site
Bengalurufull-timePosted 8mo ago

About the role

Staff Engineer to lead enhancements to search quality and ML-based relevance at Databricks, focusing on ranking, query understanding, and expanding asset coverage to enable scalable, AI-powered search experience.

Responsibilities

  • Drive the development and deployment of ML based search and discovery relevance models and systems integrated with Databricks' products and services.
  • Design and implement automated ML and NLP pipelines for data preprocessing, query understanding and rewrite, ranking and retrieval, and model evaluation, enabling rapid experimentation and iteration.
  • Collaborate with product managers and cross-functional teams to drive technology-first initiatives that enable novel business strategies and product roadmaps for the search and discovery experience.
  • Contribute to building a robust framework for evaluating search ranking improvements - both offline and online.

Qualifications

  • BS+ (M.S. or PhD preferred) in Computer Science, or a related field.
  • 10+ years experience developing search relevance systems at scale in production or in high-impact research environments.
  • Experience applying LLM to search relevance
  • Experience in one or more of the following: Query understanding, NLP, Text mining, Recommendations, Personalization, Discovery, Conversational AI
  • Strong understanding of computer science fundamentals.
  • Contributions to well-used open-source projects.

Skills

Required:ML-based search relevance modelsQuery understandingNLP pipelines

Benefits

  • Comprehensive benefits and perks; region-specific details available via provided link.
Last updated Sep 10, 2026. Interested? Apply on the company site.

About Databricks

Leading Data and AI Platform for Enterprises

Databricks offers a unified platform for data, analytics and AI. Build better AI with a data-centric approach. Simplify ETL, data warehousing, governance and AI on the Data Intelligence Platform.

AIanalyticsbig datadatadata warehousingenterprise-infrastructuregovernanceliquid-ai-recruiting-target
Company profileWebsite

More open roles at Databricks