# Staff Software Engineer - Machine Learning

**Company:** [Databricks](https://hotfix.jobs/companies/databricks)
**Location:** Bengaluru, India
**Role:** ML Engineering
**Experience:** 10+ years
**Skills:** Machine Learning, Natural Language Processing, LLMs, Search Relevance, Query Understanding, Text Mining, Information Retrieval, Recommendations, Personalization, Conversational AI, Ranking Models, Model Evaluation, Open Source
**Posted:** 2026-08-11

> Leads the development of machine-learning search relevance systems, including query understanding, ranking, retrieval, and evaluation pipelines. The role requires 10+ years of search relevance experience and expertise in NLP, LLMs, or related discovery technologies.

## Job Description

## Key Responsibilities

- Drive the development and deployment of machine-learning-based search and discovery relevance models and systems integrated with Databricks products and services.
- Design and implement automated machine learning and natural language processing pipelines for data preprocessing, query understanding and rewriting, ranking and retrieval, and model evaluation.
- Enable rapid experimentation and iteration across search and discovery systems.
- Collaborate with product managers and cross-functional teams on technology initiatives and product roadmaps for the search and discovery experience.
- Build a robust framework for evaluating search-ranking improvements, both offline and online.

## Requirements

- Bachelor's degree or higher in Computer Science or a related field; a master's degree or PhD is preferred.
- 10+ years of experience developing search relevance systems at scale in production or high-impact research environments.
- Experience applying large language models to search relevance.
- Experience in one or more of the following areas:
  - Query understanding
  - Natural language processing
  - Text mining
  - Recommendations
  - Personalization
  - Discovery
  - Conversational AI
- Strong understanding of computer science fundamentals.
- Contributions to well-used open-source projects.

## Compensation and Benefits

- Comprehensive benefits and perks are offered, with specific details varying by region.

## Similar jobs

- [Staff Software Engineer - Search Platform](https://hotfix.jobs/jobs/9c9d7c76-3610-4456-81fe-2662b6af213e) - Databricks - Bengaluru, India
- [Staff Machine Learning Engineer](https://hotfix.jobs/jobs/2cb6da8f-4a54-448d-a4b6-fc05f1412a12) - Payabli - Remote
- [Staff Software Engineer](https://hotfix.jobs/jobs/4f600fb6-7f6e-4718-b395-16f02be6a016) - Databricks - Bengaluru, India
- [Senior Software Engineer, NLP](https://hotfix.jobs/jobs/be3f0222-b08a-4620-9ddc-402209ea73f7) - GitLab - Remote
- [Senior Software Engineer , Trust](https://hotfix.jobs/jobs/db6ff47c-2411-4457-a551-f83493a3f865) - Airbnb - Bengaluru, India - ₹4.5M – ₹6.5M/yr

**Apply:** https://hotfix.jobs/jobs/b2e23245-bfe9-48d4-9338-2b865f5b19f6
**Canonical:** https://hotfix.jobs/jobs/b2e23245-bfe9-48d4-9338-2b865f5b19f6