Databricks
Staff Machine Learning Engineer
$190K–$285K
San Francisco, California +1 · Staff · Full-time · Posted Feb 2, 2026
- Location
- San Francisco, California
Mountain View, California - Experience
- 2-8 years of machine learning engineering experience in high-velocity, high-growth companies; alternatively, relevant ML research in academia is considered an equivalent qualification.
About the job
Develop and deploy machine learning models, systems, and scalable backend infrastructure that improve GenAI features across Databricks products.
What you'll do
- Shape applied AI product features and develop and deploy AI models and systems.
- Create data collection, fine-tuning, and LLM technologies, and design ML pipelines for training and evaluation.
- Build reusable GenAI backend systems, logging, telemetry, and evaluation tools in collaboration with research, engineering, and product teams.
What they ask for
- Strong track record working with language modeling technologies, including generative or embedding methods, model architectures, datasets, or evaluation benchmarks.
- Proficiency in Python, TensorFlow/PyTorch, and scalable ML architectures.
- Ability to drive end-to-end model development from research and prototyping through deployment and monitoring.
- Strong coding and software engineering skills, including familiarity with testing, code reviews, and deployment.
Nice to have
- Experience with LLM fine-tuning, prompt engineering, and retrieval-augmented generation (RAG).
Pay
$190,000 — $285,000 USD (base salary)
Tools
Python, TensorFlow/PyTorch
AI focus
Develop GenAI products and capabilities, improve LLM quality, and build models and systems for tasks such as code suggestions, text-to-SQL generation, and knowledge QA.
Model-written summary · the original posting is authoritative