Staff Machine Learning Engineer

$190k - $285k • San Francisco, California

Posted 26d ago

Remote Work Policy

On-site

Categories

Machine Learning Engineer

About the job

The Applied AI team at Databricks is at the forefront of advancing GenAI-powered products, having launched Databricks Assistant, AI/BI Genie, and Agent Bricks. These products are used by hundreds of thousands of users daily, tackling challenging problems like code suggestion, text-to-sql generation, and knowledge QA. As GenAI products evolve, we are seeking engineers to drive the next phase of development, focusing on enhancing LLM quality, expanding GenAI capabilities, and strengthening platform architecture for AI interactions at scale.

Responsibilities

  • Shape the direction of applied AI areas and intelligence features in products.
  • Drive the development and deployment of state-of-the-art AI models and systems.
  • Develop novel data collection, fine-tuning, and LLM technologies for optimal performance.
  • Design and implement ML pipelines for data preprocessing, feature engineering, model training, hyperparameter tuning, and model evaluation.
  • Work closely with cross-functional teams to deliver impactful AI solutions.
  • Build scalable, reusable backend systems to support GenAI products.
  • Develop robust logging, telemetry, and evaluation harnesses for reliable model performance.

Requirements

  • 2-8 years of machine learning engineering experience or strong relevant ML research background.
  • Strong track record working with language modeling technologies (generative/embedding techniques, model architectures, fine-tuning, evaluation).
  • Proficiency in Python, TensorFlow/PyTorch, and scalable ML architectures.
  • Ability to drive end-to-end model development from research 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, including testing, code reviews, and deployment.
  • Experience with LLM fine-tuning, prompt engineering, and retrieval-augmented generation (RAG) is a bonus.

Benefits

  • Eligibility for annual performance bonus
  • Equity
  • Health insurance
  • Paid time off

About Databricks

Get new AI jobs in your inbox

A weekly digest of the newest AI engineering roles.

© 2026 AI Job Board. All rights reserved.