Staff Software Engineer - Machine Learning (Search)
Salary not listed • Bengaluru, India • Full-time
Posted 1y ago
This job has been open for more than 60 days. It may no longer be actively hiring, so check with the company before putting a lot into an application.
About the job
The Applied AI team at Databricks is at the forefront of advancing AI/ML-powered products, focusing on enhancing search quality for customers who interact with millions of assets on the platform. In 2025, the team will concentrate on improving search ranking, query understanding, building robust evaluation frameworks, and expanding asset coverage to ensure seamless search at scale. This role offers 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.
Responsibilities
- Drive the development and deployment of ML-based search and discovery relevance models and systems.
- Design and implement automated ML and NLP pipelines for data preprocessing, query understanding, ranking, retrieval, and model evaluation.
- Collaborate with product managers and cross-functional teams to drive technology-first initiatives for search and discovery.
- Contribute to building a robust framework for evaluating search ranking improvements, both offline and online.
Requirements
- BS+ (M.S. or PhD preferred) in Computer Science or a related field.
- 10+ years of experience developing search relevance systems at scale in production or high-impact research environments.
- Experience applying LLMs 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.