Google Cloud and Nvidia announced on Tuesday they are expanding their partnership to bring new artificial intelligence (AI) infrastructure and software to customers and allow them to "build and deploy massive models for generative AI and speed data science workloads."

The companies noted that PaxML, Google's framework for building massive large language models (LLMs), has been optimized for Nvidia accelerated computing. "Originally built to span multiple Google TPU accelerator slices, PaxML now enables developers to use NVIDIA H100 and A100 Tensor Core GPUs [graphics processing unit] for advanced and fully configurable experimentation and scale. A GPU-optimized PaxML container is available immediately in the NVIDIA NGC software catalog. In addition, PaxML runs on JAX, which has been optimized for GPUs leveraging the OpenXLA compiler," they said.

Furthermore, as part of the agreement, Google will also integrate serverless Spark with Nvidia GPUs through Google's Dataproc service, which will "help data scientists speed Apache Spark workloads to prepare data for AI development."