Staff Software Engineer - AI/ML

Staff Machine Learning Engineer, CustomerLake (ML/LLM) RDQ427R109 At Databricks, we are passionate about enabling data teams to solve the world's toughest problems — from making the next mode of transportation a reality to accelerating the development of medical breakthroughs. We do this by building and running the world's best Data Intelligence Platform so our customers can use deep data insights to improve their business. Founded by engineers — and customer obsessed — we leap at every opportunity to tackle technical challenges, from designing next-gen UI/UX for interfacing with data to scaling our services and infrastructure across millions of virtual machines. And we're only getting started. As one of the first engineers in the NYC Engineering office, you'll join a small, nimble team building new products from the ground up. We're building CustomerLake, the Customer Data Platform on Databricks, to bring enterprise-grade ML and AI personalization to every company whose data already lives on Databricks. The best B2C and B2B brands have historically relied on in-house ML/AI teams to power personalization, recommendations, churn and lifetime-value modeling, and audience targeting. Our goal is to deliver that same capability to companies that don't have an in-house team but already have their data in order on Databricks. This is a true 0-to-1 environment, combining the excitement of a startup with the resources of a tech leader like Databricks. The impact you'll have: Evaluate ML and LLM approaches for CustomerLake's personalization use cases, push the models and algorithms forward, and continuously improve quality over time Go deep on how models behave in production: inspect individual traces, understand how the models reason, and tune and improve fro