Databricks Logo

Databricks

Staff Software Engineer - Machine Learning (Search)

Reposted 17 Days Ago
Be an Early Applicant
In-Office
Bengaluru, Bengaluru Urban, Karnataka, IND
Expert/Leader
In-Office
Bengaluru, Bengaluru Urban, Karnataka, IND
Expert/Leader
Lead development and deployment of ML-based search relevance models and pipelines. Improve query understanding, ranking, retrieval, and evaluation frameworks. Collaborate cross-functionally to productize search improvements and enable large-scale, robust search experiences.
The summary above was generated by AI

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 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. To learn more, follow Databricks on Twitter, LinkedIn and Facebook.
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, please visit https://www.mybenefitsnow.com/databricks. 

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.

Top Skills

Conversational Ai
Llm
Ml Pipelines
Model Evaluation
Nlp
Personalization
Query Understanding
Ranking And Retrieval
Recommendations
Text Mining

Similar Jobs

An Hour Ago
In-Office
Bangalore, Bengaluru Urban, Karnataka, IND
Junior
Junior
Artificial Intelligence • Big Data • Cloud • Information Technology • Software • Cybersecurity • Data Privacy
The Associate Accountant manages month-end close processes, ensures compliance with US GAAP, prepares financial entries, and assists with audits and reconciliations.
Top Skills: BlacklineFloqastGaapIfrsNetSuite
5 Hours Ago
Hybrid
Bangalore, Bengaluru Urban, Karnataka, IND
Senior level
Senior level
Automotive • Hardware • Robotics • Software • Transportation • Manufacturing
The Specialist Generative AI will develop and integrate generative AI solutions into enterprise applications, ensuring compliance with IT security standards and optimizing AI performance.
Top Skills: Ai Agent FrameworksAWSAzureC#Ci/CdDatabricksDockerGenerative AiKubernetesMachine LearningMlflowPython
5 Hours Ago
Hybrid
Bangalore, Bengaluru Urban, Karnataka, IND
Mid level
Mid level
Automotive • Hardware • Robotics • Software • Transportation • Manufacturing
The Operations Administrator will handle administrative activities, manage infrastructure topics, oversee PLM ticket management, and ensure seamless communication across departments, enhancing operational efficiency.
Top Skills: DmsSAP

What you need to know about the Bengaluru Tech Scene

Dubbed the "Silicon Valley of India," Bengaluru has emerged as the nation's leading hub for information technology and a go-to destination for startups. Home to tech giants like ISRO, Infosys, Wipro and HAL, the city attracts and cultivates a rich pool of tech talent, supported by numerous educational and research institutions including the Indian Institute of Science, Bangalore Institute of Technology, and the International Institute of Information Technology.

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account