
About Databricks
Empowering data teams with unified analytics
Key Highlights
- Headquartered in San Francisco, CA
- Valuation of $43 billion with $3.5 billion raised
- Serves over 7,000 customers including Comcast and Shell
- Utilizes Apache Spark for big data processing
Databricks, headquartered in San Francisco, California, is a unified data analytics platform that simplifies data engineering and collaborative data science. Trusted by over 7,000 organizations, including Fortune 500 companies like Comcast and Shell, Databricks has raised $3.5 billion in funding, ac...
🎁 Benefits
Databricks offers competitive salaries, equity options, generous PTO policies, and a remote-friendly work environment. Employees also benefit from a l...
🌟 Culture
Databricks fosters a culture of innovation with a strong emphasis on data-driven decision-making. The company values collaboration across teams and en...
Skills & Technologies
Overview
Databricks is hiring a Staff Machine Learning Engineer to drive the development of GenAI-powered products. You'll work with technologies like Python and TensorFlow to enhance AI capabilities. This position requires significant experience in machine learning and data engineering.
Job Description
Who you are
You have a strong background in machine learning and AI, with at least 5 years of experience developing and deploying machine learning models. Your expertise in Python and frameworks like TensorFlow and PyTorch allows you to build state-of-the-art AI systems that enhance product capabilities. You are familiar with data engineering principles and can design efficient data pipelines for model training and evaluation.
You thrive in collaborative environments, working closely with cross-functional teams including AI researchers and product managers. Your ability to communicate complex technical concepts to non-technical stakeholders is a key strength. You are passionate about advancing AI technologies and are eager to tackle challenging problems in the field.
What you'll do
As a Staff Machine Learning Engineer at Databricks, you will shape the direction of applied AI features in our products. You will drive the development and deployment of advanced AI models that directly impact the performance of Databricks' offerings. Your role will involve designing and implementing machine learning pipelines for data preprocessing, feature engineering, and model evaluation, enabling rapid experimentation and iteration.
You will collaborate with product teams to enhance GenAI capabilities across Databricks products, focusing on improving LLM quality and expanding functionalities. Your work will include developing novel data collection and fine-tuning techniques to achieve optimal model performance. You will also mentor junior engineers, sharing your knowledge and expertise to foster a culture of learning and innovation within the team.
What we offer
At Databricks, we offer a dynamic work environment where you can make a significant impact on the future of AI technologies. You will have the opportunity to work on cutting-edge projects that are used by hundreds of thousands of users daily. We provide competitive compensation and benefits, along with opportunities for professional growth and development. Join us in our mission to advance AI and empower organizations to harness the power of data.
Interested in this role?
Apply now or save it for later. Get alerts for similar jobs at Databricks.
Similar Jobs You Might Like
Based on your interests and this role

Machine Learning Engineer
Uber is hiring a Staff Machine Learning Engineer to lead technical initiatives in the Consumer Incentives team. You'll work on enhancing user experience through machine learning and optimization solutions. This role requires 6+ years of experience in ML engineering and expertise in frameworks like PyTorch and TensorFlow.

Machine Learning Engineer
Headway is hiring a Staff Machine Learning Engineer to lead the development of a matching system that connects patients with mental health providers. You'll work on cutting-edge machine learning technologies to enhance user experience. This role requires significant experience in machine learning and software development.

Machine Learning Engineer
Quantcast is hiring a Senior Machine Learning Engineer to develop and maintain machine learning systems that optimize advertising technology. You'll work with large datasets and advanced algorithms to enhance the company's AI-powered Demand Side Platform. This role requires strong expertise in machine learning and data analysis.

Machine Learning Engineer
Sentry is hiring a Staff Machine Learning Engineer to develop AI models and agents that enhance their software monitoring tools. You'll work with Python and machine learning frameworks like TensorFlow and Keras in San Francisco. This role requires significant experience in AI and ML.

Machine Learning Engineer
Patreon is hiring a Senior Staff Machine Learning Engineer to support their Relevance team in building ML-driven systems for search and recommendations. You'll work with data analysis, model training, and deployment in a hybrid work environment based in San Francisco or New York.