Staff Software Engineer - Machine Learning (Search)
Databricks
Bengaluru, IndiaPosted 1 month ago
Skill Required
Engineering - PipelineMachine Learning EngineerSoftware EngineerSoftware DeveloperMachine Learningrelated fieldDatabricksbuildingetc.)designUnityNLPSQLGenerative AIAI
Key highlights
- 10+ years experience required
- Focus on ML-based search and discovery relevance models
- Work with cutting-edge AI/ML models
- Collaborate with cross-functional teams
- Comprehensive benefits package
Role overview
The Applied AI team at Databricks leads the advancement of AI/ML-powered products, focusing on enhancing search quality for the platform’s vast and growing assets (tables, notebooks, dashboards, etc.). As a Staff Engineer, you will drive improvements in search ranking, query understanding, and evaluation frameworks to enable seamless search at scale, working with cutting-edge models and cross-functional teams to shape the future of AI-driven search and discovery experiences.
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.
Requirements
- 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
- Strong understanding of computer science fundamentals.
Nice to have
- Experience in one or more of the following: Query understanding, NLP, Text mining, Recommendations, Personalization, Discovery, Conversational AI
- Contributions to well-used open-source projects.
Benefits
- Comprehensive benefits and perks that meet the needs of all employees (region-specific details available via provided link).
Additional details
- P-1408
- In 2025, focus areas include enhancing search ranking, improving query understanding, building robust evals, and growing the coverage of assets to enable seamless search at scale.
- 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.
- About Databricks: Databricks is the Data and AI company. More than 20,000 organizations worldwide — including adidas, AT&T, Bayer, Block, Mastercard, Rivian, Unilever, and 70% of the Fortune 500 — rely on the Databricks Data + AI Platform to build and scale data and AI apps, analytics and agents. Headquartered in San Francisco with 30+ offices around the globe, Databricks offers a unified platform that includes Genie, Lakebase, Agent Bricks, Lakeflow, Lakehouse, and Unity Catalog.
- 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.