Staff Machine Learning Engineer

Who We Are Our Team Databricks Ventures Contact Us Careers Working at Databricks Open Jobs Press Awards and Recognition Newsroom Security and Trust Security and Trust DATA + AI SUMMIT JUNE 15–18 | SAN FRANCISCO Don’t miss our biggest Summit yet. Save 50% with early-bird pricing. Register Ready to get started? Get a Demo DATA + AI SUMMIT JUNE 15–18 | SAN FRANCISCO Don’t miss our biggest Summit yet. Save 50% with early-bird pricing. Register Login Try Databricks Overview Culture

Benefits Diversity Engineering Research Students & new grads Back to search results Staff Machine Learning Engineer San Francisco, California Apply now P-1504 The Applied AI team at Databricks sits at the forefront of advancing GenAI-powered products. Over the past years, we’ve launched Databricks Assistant , AI/BI Genie , and Agent Bricks working with product teams, and made significant strides in LLM quality for these products. These products are used by 100s of thousands of Databricks users every day. We are tackling challenging problems like code suggestion, error detection and correction, text-to-sql generation, automatic pipeline generation, knowledge QA and many others. As our GenAI products continue to evolve, we are seeking multiple GenAI Engineers from junior levels to more senior levels to drive the next phase of development. In 2025, we will focus on enhancing LLM quality, expanding GenAI capabilities across Databricks products, and strengthening our platform architecture to enable seamless AI interactions at scale. Key

Responsibilities Shape the direction of our applied AI areas and intelligence features in our products . Drive the development and deployment of state-of-the-art AI models and systems that directly impact the capabilities and performance of Databricks' products and services (e.g., Databricks Assistant and AI/BI Genie). Develop novel data collection, fine-tuning, and LLM technologies that achieve optimal performance on specific tasks and domains. Design and implement ML pipelines for data preprocessing, feature engineering, model training, hyperparameter tuning, and model evaluation, enabling rapid experimentation and iteration. Work closely with cross-functional teams, including AI researchers, ML engineers, and product teams, to deliver impactful AI solutions that enhance user productivity and satisfaction. Build scalable, reusable backend systems to support GenAI products across the company. Develop robust logging, telemetry, and evaluation harnesses to ensure reliable model performance. What We’re Looking For 2-8 years of machine learning engineering experience in high-velocity, high-growth companies. Alternatively, a strong background in relevant ML research in academia will be considered as an equivalent qualification. Strong track record of working with language modeling technologies. This could include the following: Developing generative and embedding techniques, modern model architectures, fine tuning / pre-training datasets, and evaluation benchmarks. Proficiency in Python, TensorFlow/PyTorch, and scalable ML architectures. Ability to drive end-to-end model development, from research and prototyping to deployment and monitoring. Strong analytical and problem-solving skills, with a passion for improving AI-driven user experiences. Strong coding and software engineering skills, and familiarity with software engineering principles around testing, code reviews and deployment. Experience with LLM fine-tuning, prompt engineering, and retrieval-augmented generation (RAG) is a

bonus.

Why Join Us? 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. If you're excited about pushing the boundaries of AI in real-world applications, we’d love to hear from you! Please note we are open to employees working from our Mountain View, CA office for this position. Pay Range Transparency Databricks is committed to fair and equitable

compensation practices. The pay range(s) for this role is listed below and represents the expected salary range for non-commissionable roles or on-target earnings for commissionable roles. Actual

compensation packages are based on several factors that are unique to each candidate, including but not limited to job-related skills, depth of experience, relevant certifications and training, and specific work location. Based on the factors above, Databricks anticipates utilizing the full width of the range. The total

compensation package for this position may also include eligibility for annual performance

bonus, equity, and the

benefits listed above. For more information regarding which range your location is in visit our page here . Local Pay Range $190,000 — $285,000 USD About Databricks Databricks is the data and AI company. More than 10,000 organizations worldwide — including Comcast, Condé Nast, Grammarly, and over 50% of the Fortune 500 — rely on the Databricks Data Intelligence Platform to unify and democratize data, analytics and AI. Databricks is headquartered in San Francisco, with offices around the globe and was founded by the original creators of Lakehouse, Apache Spark™, Delta Lake and MLflow. To learn more, follow Databricks on Twitter , LinkedIn and Facebook .

Benefits At Databricks, we strive to provide comprehensive

benefits and

perks that meet the needs of all of our employees. For specific details on the

benefits offered in your region, please visit https://www.mybenefitsnow.com/databricks . 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. Why Databricks Discover For Executives For Startups Lakehouse Architecture Databricks AI Research Customers Customer Stories Partners Partner Overview Partner Program Find a Partner Partner Spotlight Cloud Providers Partner Solutions Why Databricks Discover For Executives For Startups Lakehouse Architecture Databricks AI Research Customers Customer Stories Partners Partner Overview Partner Program Find a Partner Partner Spotlight Cloud Providers Partner Solutions Product Databricks Platform Platform Overview Sharing Governance Artificial Intelligence Business Intelligence Database Data Management Data Warehousing Data Engineering Data Science Application Development Pricing Pricing Overview Pricing Calculator Open Source Integrations and Data Marketplace IDE Integrations Partner Connect Product Databricks Platform Platform Overview Sharing Governance Artificial Intelligence Business Intelligence Database Data Management Data Warehousing Data Engineering Data Science Application Development Pricing Pricing Overview Pricing Calculator Open Source Integrations and Data Marketplace IDE Integrations Partner Connect Solutions Databricks For Industries Communications Financial Services Healthcare and Life Sciences Manufacturing Media and Entertainment Public Sector Retail View All Cross Industry Solutions Cybersecurity Marketing Data Migration Professional Services Solution Accelerators Solutions Databricks For Industries Communications Financial Services Healthcare and Life Sciences Manufacturing Media and Entertainment Public Sector Retail View All Cross Industry Solutions Cybersecurity Marketing Data Migration Professional Services Solution Accelerators Resources Documentation Customer Support Community Learning Training Certification Free Edition University Alliance Databricks Academy Login Events Data + AI Summit Data + AI World Tour AI Days Event Calendar Blog and Podcasts Databricks Blog Databricks AI Research Blog Data Brew Podcast Champions of Data & AI Podcast Resources Documentation Customer Support Community Learning Training Certification Free Edition University Alliance Databricks Academy Login Events Data + AI Summit Data + AI World Tour AI Days Event Calendar Blog and Podcasts Databricks Blog Databricks AI Research Blog Data Brew Podcast Champions of Data & AI Podcast About Company

Who We Are Our Team Databricks Ventures Contact Us Careers Open Jobs Working at Databricks Press Awards and Recognition Newsroom Security and Trust About Company

Who We Are Our Team Databricks Ventures Contact Us Careers Open Jobs Working at Databricks Press Awards and Recognition Newsroom Security and Trust Databricks Inc. 160 Spear Street, 15th Floor San Francisco, CA 94105 1-866-330-0121 See Careers at Databricks © Databricks 2026 . All rights reserved. Apache, Apache Spark, Spark, the Spark Logo, Apache Iceberg, Iceberg, and the Apache Iceberg logo are trademarks of the Apache Software Foundation . Privacy Notice | Terms of Use | Modern Slavery Statement | California Privacy | Your Privacy Choices

A meeting lands on your calendar

If they’re a yes, I book the chat. You show up — that’s the whole job-hunt.