Google Cloud and NVIDIA have teamed up on new AI infrastructure and software for enterprises to build and deploy massive models for generative AI and speed data science workloads.
The joint effort provides end-to-end machine learning services to make it easy to run AI supercomputers with Google Cloud offerings built on NVIDIA technologies. The hardware and software integrations utilise the same NVIDIA technologies employed over the past two years by Google DeepMind and Google research teams.
“We’re at an inflection point where accelerated computing and generative AI have come together to speed innovation at an unprecedented pace. Our expanded collaboration with Google Cloud will help developers accelerate their work with infrastructure, software and services that supercharge energy efficiency and reduce costs,” said Jensen Huang, Founder and CEO of NVIDIA.
Google’s framework for building massive large language models (LLMs), PaxML, is now optimised 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 for advanced and fully configurable experimentation and scale. A GPU-optimised PaxML container is available immediately in the NVIDIA NGC software catalogue. PaxML also runs on JAX, which has been optimised for GPUs leveraging the OpenXLA compiler.
Google DeepMind and other Google researchers are among the first to use PaxML with NVIDIA GPUs for exploratory research.
The NVIDIA-optimised container for PaxML will be available immediately on the NVIDIA NGC container registry to researchers, startups and enterprises worldwide that are building the next generation of AI-powered applications.
Google Cloud and NVIDIA have also announced Google’s integration of serverless Spark with NVIDIA GPUs through Google’s Dataproc service. This will help data scientists speed Apache Spark workloads to prepare data for AI development.
“Many of Google’s products are built and served on NVIDIA GPUs, and many of our customers are seeking out NVIDIA accelerated computing to power efficient development of LLMs to advance generative AI,” said Thomas Kurian, CEO of Google Cloud.
Photo: Google Cloud CEO Thomas Kurian (left) and NVIDIA Founder and CEO Jensen Huang at Google Cloud Next
Entelechy Asia 