Skip Navigation

PyTorch on Azure

Enterprise-ready PyTorch experience in the cloud

PyTorch is an open-source deep learning framework that accelerates the path from research to production. Data scientists at Microsoft use PyTorch as the primary framework to develop models that enable new experiences in Microsoft 365, Bing, Xbox, and more. Microsoft is a top contributor to the PyTorch ecosystem with recent contributions such as PyTorch Profiler.

PyTorch on Azure–better together


Train and deploy models reliably at scale using a built-in PyTorch environment within Azure Machine Learning and rest assured that your entire PyTorch stack is fully supported through PyTorch Enterprise.

Accelerated performance

Decrease your time to market with powerful GPU hardware, a production-grade software accelerator with ONNX Runtime and the latest innovative scaling techniques with DeepSpeed in Azure.

Strengthen the ecosystem

Achieve more with the rich PyTorch ecosystem of tools and capabilities, including PyTorch Profiler. Microsoft actively contributes to the PyTorch environment to make the experience better.

PyTorch Enterprise

Microsoft is a founding member of the PyTorch Enterprise Support Group, creating a reliable production experience with enterprise-grade support that benefits both Azure customers and the PyTorch community users. With PyTorch Enterprise, you can be confident that Azure is the best place to run PyTorch.

Read the documentation

Long-term support

Microsoft will provide commercial support for the public PyTorch codebase. Microsoft provides long-term support to selected versions of PyTorch for up to two years, enabling a stable production experience without frequent major upgrade investment.

Prioritised troubleshooting

Microsoft Premier and Unified support customers are automatically eligible for PyTorch Enterprise at no additional cost. The dedicated PyTorch team in Azure will prioritise, develop and deliver hotfixes to customers as needed.

Azure integration

The latest release of PyTorch will be included with Azure Machine Learning, along with other PyTorch add-ons including ONNX Runtime for faster inferencing. Microsoft will continue to invest to improve PyTorch inference and training speed.

Open source

PyTorch Enterprise benefits not only Azure customers but also the PyTorch community users. Selected code that aligns with PyTorch will be fed back to the public PyTorch distribution so everyone in the community can benefit.

Trusted by companies of all sizes

"Other deep learning frameworks and cloud services are out there, but we think Azure, Azure Machine Learning, and PyTorch are the best choices because they enhance accuracy, efficiency, scalability, and speed of development."

Yuji Fukaya, Manager AI Consulting Group AI Transformation Centre, Information Services International-Dentsu

"The new enterprise-level offering by Microsoft closes an important gap. Serving PyTorch models in production can be a challenge. The direct involvement of Microsoft lets us deploy new versions of PyTorch to Azure with confidence."

Jeremy Jancsary, Sr. Principal Research Scientist, Nuance

"Crayon has been using PyTorch on Azure and enjoying the smooth integration. With PyTorch Enterprise, we have more confidence to leverage the most cutting-edge features offered by newer PyTorch versions in our customers' projects."

Tailai Wen, Lead Data Scientist, Crayon

"Running PyTorch on Azure gives us the best platform to build our embodied intelligence. It's easy for our engineers to run the experiments they need, all at once, at petabyte scale."

Pablo Castellanos Garcia, VP of Engineering, Wayve

"With Azure AI and PyTorch, we combined focused applications of AI with journalistic processes and financial intelligence, yielding a solution that is unique in the market and valuable for cryptocurrency investors."

Zoiner Tejada, CEO of Solliance and CTO of Baseline

"We use Azure Machine Learning and PyTorch in our new framework to develop and move AI models into production faster, in a repeatable process that allows data scientists to work both on-premises and in Azure."

Tom Chmielenski, Principal MLOps Engineer

Microsoft is an active contributor to an ecosystem of PyTorch open-source projects

PyTorch Profiler

PyTorch Profiler is an open-source tool that helps you understand the hardware resource consumption, such as time and memory, of various PyTorch operations in your model and resolve performance bottlenecks. This makes your model execute faster and cheaper with less overhead.

ONNX Runtime on PyTorch

As deep-learning models get bigger, reducing training time becomes both a financial and environmental issue. ONNX Runtime accelerates large-scale, distributed training of PyTorch transformer models with a one-line code change. Combine with DeepSpeed to further improve training speed on PyTorch.

PyTorch on Windows

Microsoft maintains PyTorch builds for Windows so your team can enjoy well-tested and stable builds, simple and reliable installation, quickstarts and tutorials, high performance and support for more advanced features such as distributed GPU training.

ONNX Runtime: A runtime for accelerated inferencing and training of PyTorch models, supporting Windows, Mac, Linux, Android, and iOS and optimised for a variety of hardware accelerators.

DeepSpeed: A library of algorithms for training of next-generation large models, including state-of-the-art model-parallel training algorithms and other optimisations for distributed training.

Hummingbird: A library that compiles traditional models like scikit-learn or LightGBM into PyTorch tensor computation for faster inference.

Two ways to use Azure for PyTorch development

Accelerate your workflow with Azure Machine Learning

Build, train and deploy PyTorch models with ease. Azure Machine Learning removes the heavy lifting of end-to-end machine learning workflows while also handling housekeeping tasks such as data preparation and experiment tracking, which cuts time to production from weeks to hours.

Develop with preconfigured Azure Data Science Virtual Machines

Data Science Virtual Machines come installed with PyTorch, the necessary GPU drivers and a comprehensive suite of other popular data science tools. Get a frictionless development experience out of the box and the ability to work with all Azure hardware configurations including GPUs.

Watch the PyTorch Enterprise on Azure announcement at Microsoft Build 2021

Learn more about the new collaboration with Facebook, the PyTorch Enterprise Support Programme. Microsoft is launching PyTorch Enterprise on Azure to provide long-term support, prioritised troubleshooting and integration with Azure solutions.

Watch the video

Learn PyTorch fundamentals

Learn the fundamentals of deep learning with PyTorch on Microsoft Learn. This beginner-friendly learning path introduces key concepts to building machine learning models in multiple domains, including speech, vision and natural language processing.

Start the learning path

Get started with PyTorch on the AI Show

Learn the basics of PyTorch, including how to build and deploy a model and how to connect to the strong community of users.

Learn the basics of PyTorch

Get to know PyTorch concepts and modules. Learn how to load data, build deep neural networks and train and save your models in this quickstart guide.

Watch the video

PyTorch, the PyTorch logo and any related marks are trademarks of Facebook, Inc.

Accelerate your PyTorch project in the cloud with Azure