Databricks

Databricks

Posted via Greenhouse

Staff Software Engineer - Machine Learning (Search)

Posted Aug 5, 2026

Role at a glance

Salary
Not Disclosed
Location
Bengaluru, India
Work arrangement
On-site
Employment
Full-time
Experience
10+ years experience developing search relevance systems at scale in production or in high-impact research environments.
Education
BS+ (M.S. or PhD preferred) in Computer Science, or a related field.

Spotted an issue?

We’ll check it against the original posting.

Log in to report

Role Summary

AI-generated

The Applied AI team at Databricks is seeking a Staff Engineer to lead search quality enhancements for the Databricks platform, which contains large volumes of customer-created assets. The role focuses on improving search ranking, query understanding, evaluation, and asset coverage to support search and discovery at scale.

What You'll Do

  • Drive development and deployment of ML-based search and discovery relevance models and systems integrated with Databricks products and...
  • Design and implement automated ML and NLP pipelines for data preprocessing, query understanding and rewriting, ranking and retrieval,...
  • Collaborate with product managers and cross-functional teams on search and discovery technology initiatives and product roadmaps.
  • Build a framework for evaluating search ranking improvements through offline and online evaluation.

Generated from the employer's posting. Verify important details before applying.

View full posting

Qualifications

BS+ in Computer Science or a related field; 10+ years developing search relevance systems at scale; experience applying LLM to search relevance; experience in query understanding, NLP, text mining, recommendations, personalization, discovery, or conversational AI; strong understanding of computer science fundamentals; contributions to well-used open-source projects.

Required

  • M.S. or PhD in Computer Science, or a related field

Original job description

Content provided by the employer

P-1408

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 2025, 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
    • Text mining
    • Recommendations
    • Personalization
    • 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

Databricks is the Data and AI company. More than 20,000 organizations worldwide — including adidas, AT&T, Bayer, Block, Mastercard, Rivian, Unilever, and 70% of the Fortune 500 — rely on the Databricks Data + AI Platform to build and scale data and AI apps, analytics and agents. Headquartered in San Francisco with 30+ offices around the globe, Databricks offers a unified platform that includes Genie, Lakebase, Agent Bricks, Lakeflow, Lakehouse, and Unity Catalog. To learn more, follow Databricks on LinkedIn, X, YouTube, and Instagram.

Benefits

At Databricks, we strive to provide comprehensive benefits and perks that meet the needs of all of our employees. For specific details on the benefits offered in your region click here.

Our Commitment to Diversity and Inclusion

At Databricks, we are committed to fostering a diverse and inclusive culture where everyone can excel. We take great care to ensure that our hiring practices are inclusive and meet equal employment opportunity standards. Individuals looking for employment at Databricks are considered without regard to age, color, disability, ethnicity, family or marital status, gender identity or expression, language, national origin, physical and mental ability, political affiliation, race, religion, sexual orientation, socio-economic status, veteran status, and other protected characteristics.

Compliance

If access to export-controlled technology or source code is required for performance of job duties, it is within Employer's discretion whether to apply for a U.S. government license for such positions, and Employer may decline to proceed with an applicant on this basis alone.

Applicant Privacy Notice

Databricks

About the company

Databricks

Large Enterprise

Databricks is a cloud-based data platform that specializes in providing solutions for data engineering, machine learning, and analytics. Founded in 2013 by the original creators of Apache Spark, the company enables organizations to unify data processing and analysis workflows, facilitating collaboration for data professionals. Databricks offers an integrated environment for data scientists, engineers, and business analysts, streamlining the process of building and deploying AI and data-driven applications. With a strong emphasis on simplifying big data management, Databricks helps businesses harness the power of their data to drive innovation and insights.