Staff Software Engineer, Search Ranking

Databricks Inc.; Bengaluru; On-site; INR 2,000,000 - 3,000,000; Full time

Job summary

A leading AI technology company in Bengaluru is seeking a Staff Engineer to enhance their search product capabilities. The role involves leading the development of ML-based search models and collaborating across teams to drive innovative strategies. Ideal candidates will have over 10 years of experience in search relevance systems, along with advanced degrees in Computer Science. This position offers a unique opportunity to work on cutting-edge AI solutions and influence the future of AI-driven products.

Qualifications

• 10+ years of experience developing search relevance systems at scale.
• Experience applying LLM to search relevance.
• Strong understanding of computer science fundamentals.

Responsibilities

• Drive the development of ML based search and discovery relevance models.
• Design and implement automated ML and NLP pipelines.
• Collaborate with product managers and teams for technology initiatives.
• Contribute to a robust framework for evaluating search ranking improvements.

Skills

Search relevance systems

Machine Learning

Natural Language Processing (NLP)

Query understanding

Conversational AI

Education

BS in Computer Science

MS or PhD preferred

Job description

The Applied AI team at Databricks sits at the forefront of advancing AI/ML-powered products. Databricks’ customers are continuously creating new assets (tables, notebooks, dashboards, datarooms, pipelines, sql queries, ml models etc.) on the platform. Some of them can have hundreds of millions of assets. Finding an asset is a critical user journey for Databricks’ customers which helps them accomplish their tasks.

As our Search product continues to evolve, we are seeking a Staff Engineer to lead enhancements to our Search Quality. In 2026, we will focus on enhancing search ranking, improving query understanding, building robust evals and growing the coverage of assets to enable seamless search at scale.

Key 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.

What We’re Looking For

• 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
• Recommendations
• Discovery
• Conversational AI
• Strong understanding of computer science fundamentals.
• Contributions to well-used open-source projects.

Why Join Us?

At Databricks, we are building state-of-the-art AI solutions that redefine how users interact with data and our products. You’ll have the opportunity to shape the future of AI-driven products at Databricks, work with cutting-edge models, and collaborate with a world-class team of AI and ML experts.

If you're excited about pushing the boundaries of AI in real-world applications, we’d love to hear from you!

About Databricks