Databricks

Staff Machine Learning Engineer

Databricks
hybrid staff full-time $190,000-$285,000 USD San Francisco, California
Apply →

First indexed 18 Apr 2026

Description

We are seeking multiple GenAI Engineers from junior levels to more senior levels to drive the next phase of development in our Applied AI team. As our GenAI products continue to evolve, 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.
  • 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.

What We’re Looking For:

  • 2-8 years of machine learning engineering experience in high-velocity, high-growth companies.
  • Strong track record of working with language modeling technologies.
  • 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.

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.

Pay Range Transparency

Databricks is committed to fair and equitable compensation practices. The pay range for this role is $190,000-$285,000 USD.

This listing is enriched and indexed by YubHub. To apply, use the employer's original posting: https://job-boards.greenhouse.io/databricks/jobs/8401114002