Staff Machine Learning Engineer

The shape of it

In the posting’s own words

As our GenAI products continue to evolve, we are seeking multiple GenAI Engineers from junior levels to more senior levels to drive the next phase of development. In 2025, 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.

What it asks for · 6

• 2-8 years of machine learning engineering experience in high-velocity, high-growth companies. Alternatively, a strong background in relevant ML research in academia will be considered as an equivalent qualification.
• Strong track record of working with language modeling technologies. This could include the following: Developing generative and embedding techniques, modern model architectures, fine tuning / pre-training datasets, and evaluation benchmarks.
• 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 coding and software engineering skills, and familiarity with software engineering principles around testing, code reviews and deployment.
• Experience with LLM fine-tuning, prompt engineering, and retrieval-augmented generation (RAG) is a bonus.

What the job covers

• 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 (e.g., Databricks Assistant and AI/BI Genie).
• 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. Develop robust logging, telemetry, and evaluation harnesses to ensure reliable model performance.

Tools and skills named

Words the posting leans on

• products12×
• model7×
• engineering5×
• genai4×
• llm4×
• user4×
• architectures3×
• data3×
• deployment3×
• development3×
• drive3×
• evaluation3×
• experience3×
• generation3×
• performance3×
• ai-driven2×

Counted from the posting after the mission statement and the legal notices are set aside. The ones near the top are the ones a screener is looking for.

More open at Databricks

• ソリューションアーキテクト (プリセールス)Tokyo, Japan
• デリバリーソリューションアーキテクトTokyo, Japan
• Account ExecutiveBengaluru, India; Mumbai, India
• Account Executive - FederalAustralian Capital Territory, Australia
• Account Executive, SingaporeSingapore
• Accounting ManagerSan Francisco, California