← All roles

Staff Machine Learning Engineer

Databricks· Engineering
Apply on company site
Mountain View, CA$190k - $285kfull-timePosted 7mo ago

About the role

Applied AI GenAI-focused engineering role at Databricks, targeting development and deployment of state-of-the-art AI models and systems to enhance Databricks products (e.g., Databricks Assistant, AI/BI Genie) and scale GenAI capabilities across the platform.

Responsibilities

  • Shape the direction of applied AI areas and intelligence features in products; drive development and deployment of state-of-the-art AI models and systems.
  • Develop novel data collection, fine-tuning, and LLM technologies for task- and domain-specific performance.
  • Design and implement ML pipelines for data preprocessing, feature engineering, model training, hyperparameter tuning, and model evaluation.
  • Collaborate with AI researchers, ML engineers, and product teams to deliver impactful AI solutions.
  • Build scalable backend systems to support GenAI products; develop robust logging, telemetry, and evaluation harnesses.

Qualifications

  • 2-8 years of machine learning engineering experience in high-velocity, high-growth companies (or strong academic ML research background).
  • Strong track record with language modeling technologies (generative and embedding techniques, modern architectures, fine-tuning/pre-training datasets, evaluation benchmarks).
  • 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; passion for improving AI-driven user experiences.
  • Strong coding and software engineering skills; familiarity with testing, code reviews, and deployment."],
  • benefits include_or_is_available_includes_details_extracted_from_description_? (not provided)
  • benefitsК?
  • domains':[ ]}
  • domains_users

Skills

Required:Machine learning engineeringLanguage modelingPythonTensorFlow/PyTorchEnd-to-end ML developmentSoftware engineering practices
Nice to have:RAG / retrieval-augmented generation

Benefits

  • Benefits details available via company link. (See job description for benefits reference.)
Last updated Sep 10, 2026. Interested? Apply on the company site.

About Databricks

Leading Data and AI Platform for Enterprises

Databricks offers a unified platform for data, analytics and AI. Build better AI with a data-centric approach. Simplify ETL, data warehousing, governance and AI on the Data Intelligence Platform.

AIanalyticsbig datadatadata warehousingenterprise-infrastructuregovernanceliquid-ai-recruiting-target
Company profileWebsite

More open roles at Databricks