Tag: Jax
-
Cloud Blog: Speed up checkpoint loading time at scale using Orbax on JAX
Source URL: https://cloud.google.com/blog/products/compute/unlock-faster-workload-start-time-using-orbax-on-jax/ Source: Cloud Blog Title: Speed up checkpoint loading time at scale using Orbax on JAX Feedly Summary: Imagine training a new AI / ML model like Gemma 3 or Llama 3.3 across hundreds of powerful accelerators like TPUs or GPUs to achieve a scientific breakthrough. You might have a team of powerful…
-
Cloud Blog: Google Cloud at GTC: A4 VMs now generally available, A4X VMs in preview
Source URL: https://cloud.google.com/blog/products/compute/google-cloud-goes-to-nvidia-gtc/ Source: Cloud Blog Title: Google Cloud at GTC: A4 VMs now generally available, A4X VMs in preview Feedly Summary: At Google Cloud, we’re thrilled to return to NVIDIA’s GTC AI Conference in San Jose CA this March 17-21 with our largest presence ever. The annual conference brings together thousands of developers, innovators,…
-
Cloud Blog: Introducing A4X VMs powered by NVIDIA GB200 — now in preview
Source URL: https://cloud.google.com/blog/products/compute/new-a4x-vms-powered-by-nvidia-gb200-gpus/ Source: Cloud Blog Title: Introducing A4X VMs powered by NVIDIA GB200 — now in preview Feedly Summary: The next frontier of AI is reasoning models that think critically and learn during inference to solve complex problems. To train and serve this new class of models, you need infrastructure with the performance and…