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 Office 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
Production-ready
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.
Long-term support
Microsoft provides long-term support (LTS) to selected versions of PyTorch for up to two years, enabling a stable production experience without frequent major upgrade investment.
Prioritized troubleshooting
Microsoft Premier support customers are automatically eligible for PyTorch Enterprise. They can request hotfixes directly to the dedicated Azure team. The requests will be prioritized, quickly addressed, and deployed in the next version of PyTorch.
Azure integration
The latest release of PyTorch will be integrated with Azure Machine Learning, along with other PyTorch add-ons including ONNX Runtime for faster inferencing.
Open source
All code is fed back to the public PyTorch distribution so everyone in the PyTorch community can benefit.
AstraZeneca
Gavin Edwards, Machine Learning Engineer, AstraZeneca"Scientists have to comb through massive amounts of data to deliver life-changing medicines. Microsoft and PyTorch are helping global biopharmaceutical company AstraZeneca to accelerate its drug discovery research."
Microsoft is an active contributor to an ecosystem of PyTorch open-source projects:
PyTorch Profiler
Introducing PyTorch Profiler, the new and improved performance debugging tool. Developed as part of a collaboration between Microsoft and Facebook, PyTorch Profiler is an open-source tool that enables accurate and efficient performance analysis and troubleshooting for large-scale deep learning models.
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 that your team can enjoy well-tested and stable builds, simple and reliable installation, quickstarts and tutorials, and high performance and support for more advanced features, such as distributed GPU training.
ONNX Runtime: 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, such as 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 as well as 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.
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.
DeepSpeed – PyTorch Developer Day
In this talk, Yuxiong He, a partner research manager at Microsoft, presented DeepSpeed, an open-source deep-learning training optimisation library compatible with PyTorch.
PyTorch on Windows
Maxim Lukiyanov, a product manager for the Azure AI platform, described improvements within the Windows platform support made in PyTorch version 1.7.
Deep learning fundamentals with PyTorch
See how to use PyTorch to solve a simple image classification problem.
Learn the basics with PyTorch
Get to know PyTorch concepts and modules. Learn how to load data, build deep neural networks, train and save your models in this quickstart guide.
Learn more about PyTorch on Azure
Blogs
PyTorch, the PyTorch logo and any related marks are trademarks of Facebook, Inc.
Accelerate your PyTorch project in the cloud with Azure
