Staff Software Engineer , Machine Learning (Search)

Databricks(9 months ago)

HybridFull TimeSenior$162,958 - $218,696 (estimated)Applied AI

About this role

The Staff Engineer on Databricks' Applied AI team will lead enhancements to Search Quality, focusing on search ranking, query understanding, robust evaluation frameworks, and scaling asset coverage. The role works closely with product managers and cross-functional teams to integrate ML- and NLP-driven search and discovery capabilities across the platform. This position helps shape strategic product roadmaps and drive AI-driven user experiences at scale.

View Original Listing

Required Skills

Search Relevance
ML Pipelines
NLP
Query Understanding
Ranking
Retrieval
Model Evaluation
LLMs
Text Mining
Recommendations

+4 more

Qualifications

BS in Computer Science
MS in Computer Science (Preferred)
PhD in Computer Science (Preferred)

About Databricks

databricks.com

Databricks is a unified Data Intelligence Platform that helps organizations build, deploy, and manage data, analytics, and AI at scale. Built on Apache Spark and Delta Lake, it provides a “lakehouse” architecture that combines data engineering, ETL, data warehousing, streaming, and machine learning (including MLflow and feature store) in a single managed service. The platform includes collaborative notebooks, scalable compute across major clouds, and built‑in governance, security, and performance optimizations. Enterprises use Databricks to simplify data pipelines, accelerate model development with a data‑centric approach, and operationalize analytics and AI.