Tag: PyTorch

Official Blog PyTorch TPU Feb. 3, 2025

Improving model performance with PyTorch/XLA 2.6 - PyTorch/XLA 2.6 improves AI model performance on Cloud TPUs. It introduces an experimental scan operator for faster compilation, host offloading to fit larger models on fewer TPUs, an alternative base Docker image with improved goodput, and reorganized documentation for easier navigation.

Official Blog PyTorch Tutorial Jan. 13, 2025

The PyTorch developer's guide to JAX fundamentals - This article provides a straightforward tutorial to help PyTorch users understand the basics of JAX by connecting its new concepts to the PyTorch building blocks they’re already familiar with.

AI Official Blog PyTorch Nov. 4, 2024

PyTorch/XLA 2.5: vLLM support and an improved developer experience - PyTorch/XLA 2.5 is now available with a set of improvements to enhance the developer experience. The release includes a clarified proposal for deprecating the older torch_xla API, improvements to the torch_xla.compile function for better debugging, and experimental support for vLLM on TPUs.

AI Official Blog PyTorch Aug. 5, 2024

Announcing PyTorch/XLA 2.4: A better Pallas and developer experience, plus “eager mode” - PyTorch/XLA 2.4 introduces several improvements to enhance developer experience and model performance. It includes updates to Pallas, a custom kernel language supporting TPUs and GPUs, with added support for Flash Attention, Paged Attention, and Megablocks' block sparse kernels.

AI Cloud Storage Official Blog PyTorch May 6, 2024

Introducing Dataflux Dataset for Cloud Storage to accelerate PyTorch AI training

GPU Official Blog PyTorch April 29, 2024

Announcing PyTorch/XLA 2.3: Distributed training, dev improvements, and GPUs

Official Blog PyTorch Vertex AI May 22, 2023

Deploying your Generative AI model in only four steps with Vertex AI and PyTorch - Use TorchServe, Cloud Storage, VertexAI and PyTorch to make your own large model for scalable, production-ready AI.

AI Machine Learning PyTorch Vertex AI May 1, 2023

Batch/Online Predictions with Pytorch Hugging Face Models on Google Cloud - Build your personalized container that adapts to your needs.

Batch Official Blog PyTorch April 17, 2023

Rapidly deploy PyTorch applications on Batch using TorchX - Batch and TorchX simplify the development and execution of PyTorch applications in the cloud to accelerate training, research, and support for ML pipelines.

Machine Learning PyTorch Vertex AI April 10, 2023

Training and serving PyTorch models in the Google Cloud with Vertex AI pipelines - Using PyTorch with Vertex AI.

Official Blog PyTorch Vertex AI March 20, 2023

Optimize PyTorch training performance with Reduction Server on Vertex AI - Data parallelism can improve your PyTorch training time and help with performance.

Machine Learning PyTorch Vertex AI March 6, 2023

Your Turn PyTorch! - Using Vertex AI to train PyTorch model.

Machine Learning Official Blog PyTorch Vertex AI Feb. 20, 2023

Serving PyTorch models with prebuilt containers on Vertex AI - How Google Cloud makes it easier for PyTorch users to deploy models in the cloud using Vertex AI.

Official Blog PyTorch TPU Oct. 10, 2022

Building Large Scale Recommenders using Cloud TPUs - In this blog post, we introduce concepts to generate and analyze traces to debug PyTorch training performance on TPU VM.

Data Science Machine Learning PyTorch Vertex AI June 27, 2022

Training a PyTorch Model on GCP Vertex AI - Training models with managed notebooks or custom training jobs.

AI Machine Learning Official Blog PyTorch Feb. 28, 2022

PyTorch on Google Cloud: Blog series recap - This blog post has a list of all the posts published as part of PyTorch on Google Cloud blog series.

Official Blog PyTorch Vertex AI Feb. 21, 2022

Orchestrating PyTorch ML Workflows on Vertex AI Pipelines - In this blog post, we show how to build and orchestrate ML pipelines for training and deploying PyTorch models on Google Cloud Vertex AI using Vertex AI Pipelines.

Machine Learning Official Blog PyTorch Jan. 24, 2022

PyTorch/XLA: Performance debugging on Cloud TPU VM: Part III - In this blog post, we introduce concepts to generate and analyze traces to debug PyTorch training performance on TPU VM.

Machine Learning Official Blog PyTorch Jan. 17, 2022

PyTorch/XLA: Performance debugging on Cloud TPU VM: Part II - In this blog post, we build upon the concepts introduced in part-1 and apply these to analyze and improve performance for a case study involving a model graph with dynamic shapes.

Machine Learning Official Blog PyTorch TPU Jan. 10, 2022

PyTorch/XLA: Performance debugging on TPU-VM part 1 - We present the fundamentals of the performance analysis of PyTorch on Cloud TPUs and discuss a performance analysis case study.

AI Official Blog PyTorch Vertex AI Sept. 20, 2021

PyTorch on Google Cloud: How to deploy PyTorch models on Vertex AI - In this blog post, we show how you can deploy PyTorch models on Google Cloud Vertex AI with the Vertex Prediction service to serve predictions from the trained model artifacts.