Want to professionalize your AI skills, pivot to an AI role and increase your salary?
Master AI Engineering with the most practical and comprehensive LLM Development certifications at Towards AI Academy.

Databricks

Staff Software Engineer - Machine Learning (Search)

Databricks

Published 11 May 2026
Bangalore, India
Full Time

Share this job

Role Highlights

Languages used

SQL

Key skills

Conversational AI
Computer Science
Product Management
Staff Engineer
LLMs
Machine Learning
Search
Deployment
NLP
PhD
Research

Tools, Libraries and Frameworks

DataBricks
MLFlow
Apache Spark
NoteBooks

Description

The role involves leading enhancements to search quality within the platform. The individual will drive the development and deployment of machine learning-based search and discovery relevance models. Responsibilities include designing and implementing automated pipelines for data preprocessing, query understanding, and model evaluation. The position requires collaboration with product managers and cross-functional teams to advance technology-first initiatives. Additionally, the role focuses on building frameworks for evaluating search ranking improvements in both offline and online environments.

Required Qualifications and Skills

Candidates must possess a Bachelor of Science degree or higher, with a Master of Science or PhD preferred in Computer Science or a related field. A minimum of 10 years of experience in developing search relevance systems at scale is required. Applicants should demonstrate expertise in applying LLMs to search relevance and possess knowledge in areas such as NLP, text mining, and conversational AI. Furthermore, a strong understanding of computer science fundamentals and a history of contributions to open-source projects are necessary.

Disclaimer

Disclaimer: Job and company description information and some of the data fields may have been generated via GPT-4 summarisation and could contain inaccuracies. The full external job listing link should always be relied on for authoritative information.

About the company

Databricks

Size

8632

HQ

San Francisco, US

Public/Private

Privately Held

Description

Databricks is the data and AI company. More than 10,000 organizations worldwide including Comcast, Condé Nast, Grammarly, and over 50% of the Fortune 500 rely on the Databricks Data Intelligence Platform to unify and democratize data, analytics and AI. Databricks is headquartered in San Francisco, with offices around the globe and was founded by the original creators of Lakehouse, Apache Spark, Delta Lake and MLflow. --- Attention: Databricks applicants Due to reports of phishing, were requesting that all Databricks applicants apply through our official Careers page at databricks.com/company/careers (good news you are here). All official communication from Databricks will come from email addresses ending with @databricks.com or @goodtime.io (our meeting tool).

Share

Share this job

Related jobs

Apache Spark
Big Data
Data Science
Solution Architect
Zurich, Switzerland
Full Time
Apache Spark
Customer Success
GCP
Data
Stockholm, Sweden
Full Time
Apache Spark
Open Source
GCP
Data
London, UK
Full Time