# Staff Machine Learning Engineer

**Company:** [Databricks](https://hotfix.jobs/companies/databricks)
**Location:** San Francisco, CA, Mountain View, CA
**Role:** ML Engineering
**Salary:** $190k – $285k/yr
**Skills:** Python, PyTorch, TensorFlow, LLMs, Fine-Tuning, Machine Learning, Prompt Engineering, RAG, Ml Pipelines, Generative AI
**Posted:** 2026-02-02

> Develops and deploys state-of-the-art GenAI models and systems for Databricks products like Assistant and Genie. Requires 2-8 years ML engineering experience, proficiency in Python/PyTorch/TensorFlow, and expertise in LLMs.

## Job Description

## Key Responsibilities
- Shape applied AI direction and intelligence features in products. Drive development and deployment of state-of-the-art AI models and systems (e.g., Databricks Assistant, AI/BI Genie).
- Develop novel data collection, fine-tuning, and LLM technologies for optimal task/domain performance.
- Design and implement ML pipelines for data preprocessing, feature engineering, model training, hyperparameter tuning, and evaluation.
- Collaborate with AI researchers, ML engineers, and product teams to deliver impactful AI solutions.
- Build scalable backend systems, logging, telemetry, and evaluation harnesses for GenAI products.

## What We’re Looking For
- 2-8 years of machine learning engineering experience in high-velocity companies, or equivalent ML research background.
- Strong track record with language modeling: generative/embedding techniques, modern architectures, fine-tuning/pre-training datasets, evaluation benchmarks.
- Proficiency in **Python**, **TensorFlow/PyTorch**, and scalable ML architectures.
- End-to-end model development from research/prototyping to deployment/monitoring.
- Strong analytical/problem-solving skills and passion for AI-driven user experiences.
- Strong coding/software engineering skills with testing, code reviews, deployment.

**Bonus:** LLM fine-tuning, prompt engineering, retrieval-augmented generation (RAG).

## Similar roles

- [Staff Software Engineer, AI Runtime](https://hotfix.jobs/jobs/2ea7944c-45ee-4f48-8c85-a9ae03a53b15) - Databricks - Mountain View, CA - $190k – $265k/yr
- [Staff Machine Learning Engineer, Computer Vision](https://hotfix.jobs/jobs/2969c5c0-8479-43bb-b9b6-f8df93b528d7) - Pinterest - San Francisco, CA - $189k – $390k/yr
- [Staff Machine Learning Engineer](https://hotfix.jobs/jobs/3c08f024-1b6c-4c43-86de-db177792a225) - Pinterest - San Francisco, CA - $189k – $390k/yr
- [Staff Software Engineer - GenAI Performance and Kernel](https://hotfix.jobs/jobs/7d20b86b-17a6-4dbe-8bc0-89703822b0fc) - Databricks - San Francisco, CA - $191k – $233k/yr
- [Staff Software Engineer - GenAI inference](https://hotfix.jobs/jobs/c24ce5c3-c403-4ff3-90c0-c110271f0098) - Databricks - San Francisco, CA - $191k – $233k/yr

**Apply:** https://hotfix.jobs/jobs/d1943f91-7f1d-4c37-840b-24ef6a11d96b
**Canonical:** https://hotfix.jobs/jobs/d1943f91-7f1d-4c37-840b-24ef6a11d96b