Databricks broadened its partnership with Amazon Web Services in a move that will put Databricks Mosaic AI on AWS for custom models. In addition, Databricks will use AWS' Trainium chips as its preferred infrastructure for model training.

Under the partnership, Databricks and AWS will give joint customers the ability to use Mosaic AI to pretrain, tune and serve large language models (LLMs) on AWS.

The two companies will also provide new integrations for Databricks on AWS Marketplace, which is becoming a vehicle for enterprise software sales. For instance, Databricks' AWS business has passed $1 billion annual revenue run rate due to AWS Marketplace.

Key items in the collaboration pact include:

  • Databricks, which recently expanded Mosaic AI to include Mosaic AI Model Serving, will support multiple model providers via Amazon Bedrock.
  • Mosaic AI models can scale up on AWS Trainium chips.
  • Joint customers can launch AI applications while keeping their data secure.
  • The two companies will expand go-to-market efforts and joint offerings. Databricks and AWS will develop custom models built with Mosaic AI on AWS Trainium. The companies also said they will collaborate with systems integrators to migrate data stores from on-premises to AWS.
  • Databricks and AWS will launch industry genAI accelerators for media and entertainment and financial services with more on deck.
  • AWS Marketplace integrations with Databricks will simplify onboarding and configuration and serverless compute.

Related: