Staff Software Engineer - Machine Learning (Search)

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

Job summary

Databricks in Bengaluru is looking for a seasoned professional with over 10 years of experience in developing ML-based search relevance systems. The candidate should hold a BS+, M.S., or Ph.D. in Computer Science or a related field, and be skilled in query understanding, NLP, and text mining. The role involves driving technology initiatives and collaborating with cross-functional teams to enhance the search experience. Databricks values diversity and inclusion, ensuring a welcoming workplace for all.

Qualifications

• 10+ years of experience developing search relevance systems at scale.
• Experience applying LLM to search relevance.
• Contributions to well-used open-source projects.

Responsibilities

• Drive the development and deployment of ML-based relevance models.
• Design and implement automated ML and NLP pipelines.
• Collaborate with cross-functional teams on product roadmaps.
• Contribute to a framework for evaluating search ranking improvements.

Skills

Development of ML-based search systems

NLP and query understanding

Text mining

Recommendations and personalization

Conversational AI

Education

BS+ (M.S. or Ph.D.) in Computer Science or related field

Job description

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 Ph.D.) in Computer Science or a related field.
• 10+ years of 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.

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.

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.