Select your cookie preferences

We use essential cookies and similar tools that are necessary to provide our site and services. We use performance cookies to collect anonymous statistics, so we can understand how customers use our site and make improvements. Essential cookies cannot be deactivated, but you can choose “Customize” or “Decline” to decline performance cookies.

If you agree, AWS and approved third parties will also use cookies to provide useful site features, remember your preferences, and display relevant content, including relevant advertising. To accept or decline all non-essential cookies, choose “Accept” or “Decline.” To make more detailed choices, choose “Customize.”

Skip to main content

PyTorch on AWS

A highly performant, scalable, and enterprise-ready PyTorch experience on AWS

Create an AWS account
Get started with PyTorch on SageMaker

Benefits

With Amazon EC2 instances, Amazon SageMaker, and PyTorch libraries.

To production scale deployments using PyTorch libraries.

Using fully managed or self-managed AWS machine learning (ML) services.

Use cases

Use PyTorch Distributed Data Parallel (DDP) systems to train large language models with billions of parameters.

Learn more

Scale inference using SageMaker and Amazon EC2 Inf1 instances to meet your latency, throughput, and cost requirements.

Learn more

Use PyTorch multimodal libraries to build custom models for use cases such as real-time handwriting recognition.

Learn more