Staff Machine Learning Engineer, CustomerLake (ML/LLM)
Databricks9 hours ago
Base Salary
$190k - $253k/yr
Responsibilities
- Evaluate ML and LLM approaches for CustomerLake personalization use cases and continuously improve model quality.
- Inspect production model traces and behavior to tune and improve models.
- Build the platform and evaluation framework for optimizing personalization against business outcomes.
- Identify and pursue novel methods and technical directions for ML/AI personalization.
- Partner with product management, engineering, and design to turn ambiguous customer problems into scalable, trustworthy solutions.
- Establish technical foundations and best practices for ML/AI personalization as the product area grows.
Requirements
- 10+ years of engineering experience with a strong foundation in shipping and improving ML/AI products.
- Hands-on experience building and evaluating ML models and/or LLM systems for real product or business use cases.
- Experience with personalization based on customer behavior or transactions, including recommendations, targeting, churn, or lifetime-value modeling.
- Proficiency in Python and modern ML frameworks such as PyTorch, including model evaluation and production AI-quality monitoring.
- Familiarity with LLMs and generative AI techniques including retrieval-augmented generation, prompt design, fine-tuning, and evaluation.
- Product mindset and ability to translate ambiguous customer problems into MVPs and iterate using data and user feedback.
- High ownership and ability to drive projects from idea through launch and adoption in 0-to-1 environments.
- Experience in martech and analytical go-to-market or business use cases is preferred.
- Academic or research experience supporting innovation and development of novel methods is preferred.
Benefits
- Databricks offers comprehensive benefits and perks, with specific offerings dependent on the employee’s region.
Categories
About Databricks
Databricks builds a cloud-based data and AI platform centered on the lakehouse architecture, combining data engineering, analytics, and machine learning with Apache Spark, Delta Lake, and MLflow. It sells subscriptions and cloud services to enterprises that need to unify data pipelines and develop large-scale AI and analytics. Founded in 2013 by the creators of Apache Spark and headquartered in San Francisco, the company is privately held and serves organizations across many industries.