Staff Machine Learning Engineer CustomerLake ML LLM
Databricks is a data and AI platform that lets organizations build analytics, AI agents, and applications on a unified, governed lakehouse.
Maintainer signals as of 9/23/2026
Funding history
Investors
About Databricks
Data engineers, analysts, and AI teams use Databricks to process large datasets, build reliable pipelines, and train models on a single governed platform. Users can run SQL analytics, serve ML predictions in real time, and deploy AI agents grounded in enterprise data. Its open lakehouse architecture provides consistent security and governance across analytical and operational workloads.
Skills
Candidate Availability
Required and preferred rules are kept separate and reflect the wording in the original posting.
About the Role
You will evaluate and improve ML and LLM approaches for personalization use cases. You will build evaluation platforms, inspect production model behavior, optimize models for business outcomes, and translate ambiguous customer problems into scalable solutions.
Requirements
- 10+ years of engineering experience shipping and improving ML or AI products
- Experience building and evaluating ML models or LLM systems for product or business use cases
- Experience with customer-behavior personalization or transaction-based modeling
- Python
- PyTorch
- Model evaluation
- Production AI quality monitoring
- LLM
- Generative AI
- Retrieval-augmented generation
- Prompt design
- Fine-tuning
- Product mindset
Responsibilities
- Evaluate ML and LLM approaches for personalization use cases
- Improve model and algorithm quality over time
- Inspect production model traces and tune model behavior
- Build platforms and evaluation frameworks for business-value optimization
- Identify and pursue novel ML and AI methods
- Partner with product management, engineering, and design on scalable solutions
- Set technical foundations and best practices for ML and AI personalization
Benefits
- Eligibility for annual performance bonus
- Equity