Optimizing Production PyTorch Models’ Performance with Graph Transformations Blog Optimizing Production PyTorch Models’ Performance with Graph Transformations 1. Introduction PyTorch supports two execution modes : eager mode and graph mode. In eager…Jade Nie, CK Luk, Xiaodong Wang, Jackie (Jiaqi) XuNovember 28, 2022
Efficient Multi-Objective Neural Architecture Search with Ax Blog Efficient Multi-Objective Neural Architecture Search with Ax tl;dr Multi-Objective Optimization in Ax enables efficient exploration of tradeoffs (e.g. between model performance and…David Eriksson, Max BalandatNovember 22, 2022
Scaling Multimodal Foundation Models in TorchMultimodal with Pytorch Distributed Blog Scaling Multimodal Foundation Models in TorchMultimodal with Pytorch Distributed Introduction In recent years, scaling model sizes has become a promising area of research. In…Ankita De, Edward Wang (EcoF), Rohan Varma, Anjali Sridhar, Kartikay KhandelwalNovember 21, 2022
Introducing TorchMultimodal – a library for accelerating exploration in Multimodal AI Blog Introducing TorchMultimodal – a library for accelerating exploration in Multimodal AI We are announcing TorchMultimodal Beta, a PyTorch domain library for training SoTA multi-task multimodal models…Kartikay Khandelwal, Ankita DeNovember 17, 2022
PyTorch Enterprise Support Program Update Blog PyTorch Enterprise Support Program Update On May 25, 2021, we announced the PyTorch Enterprise Support Program (ESP) that enabled providers to develop…PyTorch FoundationNovember 10, 2022
Extending TorchVision’s Transforms to Object Detection, Segmentation & Video tasks Blog Extending TorchVision’s Transforms to Object Detection, Segmentation & Video tasks Note: A previous version of this post was published in November 2022. We have updated…Philip Meier, Victor Fomin, Vasilis Vryniotis, Nicolas HugNovember 3, 2022
New Library Updates in PyTorch 1.13 Blog New Library Updates in PyTorch 1.13 Summary We are bringing a number of improvements to the current PyTorch libraries, alongside the…PyTorch FoundationOctober 28, 2022
PyTorch 1.13 release, including beta versions of functorch and improved support for Apple’s new M1 chips. Blog PyTorch 1.13 release, including beta versions of functorch and improved support for Apple’s new M1 chips. We are excited to announce the release of PyTorch® 1.13 (release note)! This includes Stable versions…PyTorch FoundationOctober 28, 2022
PyTorch’s Tracing Based Selective Build Blog PyTorch’s Tracing Based Selective Build Introduction TL;DR: It can be challenging to run PyTorch on mobile devices, SBCs (Single Board…Dhruv Matani, Suraj SubramanianOctober 17, 2022
Scaling PyTorch models on Cloud TPUs with FSDP Blog Scaling PyTorch models on Cloud TPUs with FSDP Introduction The research community has witnessed a lot of successes with large models across NLP,…Ronghang Hu, Vaibhav Singh, Jack Cao, Milad Mohammadi, Yeounoh Chung, Shauheen Zahirazami, Ross GirshickOctober 13, 2022
Performance Debugging of Production PyTorch Models at Meta Blog Performance Debugging of Production PyTorch Models at Meta 1. Meta’s AI Performance Profiling (MAIProf) Figure 1: A simplified illustration of the Meta’s AI…CK Luk, Lei TianSeptember 29, 2022
Announcing PyTorch Conference 2022 Blog Announcing PyTorch Conference 2022 We are excited to announce that the PyTorch Conference returns in-person as a satellite event…PyTorch FoundationSeptember 26, 2022
PyTorch strengthens its governance by joining the Linux Foundation Blog PyTorch strengthens its governance by joining the Linux Foundation Today, I am proud to announce that PyTorch is moving to the Linux Foundation (LF) as a…Soumith ChintalaSeptember 12, 2022
Fast Beam Search Decoding in PyTorch with TorchAudio and Flashlight Text Blog Fast Beam Search Decoding in PyTorch with TorchAudio and Flashlight Text Beam search decoding with industry-leading speed from Flashlight Text (part of the Flashlight ML framework) is now available with…Caroline Chen, Jacob Kahn (@jacob_d_kahn)August 29, 2022
Introducing nvFuser, a deep learning compiler for PyTorch Blog Introducing nvFuser, a deep learning compiler for PyTorch nvFuser is a Deep Learning Compiler for NVIDIA GPUs that automatically just-in-time compiles fast and…Christian Sarofeen, Piotr Bialecki, Jie Jiang, Kevin Stephano, Masaki Kozuki, Neal Vaidya, Stas BekmanAugust 26, 2022
Accelerating PyTorch Vision Models with Channels Last on CPU Blog Accelerating PyTorch Vision Models with Channels Last on CPU Overview Memory formats has significant impact on performance when running vision models, generally Channels Last…Mingfei Ma (Intel), Vitaly Fedyunin (Meta), Wei Wei (Meta)August 24, 2022
Easily list and initialize models with new APIs in TorchVision Blog Easily list and initialize models with new APIs in TorchVision TorchVision now supports listing and initializing all available built-in models and weights by name. This…Vasilis Vryniotis and Laurence RouesnelAugust 18, 2022
Empowering PyTorch on Intel® Xeon® Scalable processors with Bfloat16 Blog Empowering PyTorch on Intel® Xeon® Scalable processors with Bfloat16 Overview Recent years, the growing complexity of AI models have been posing requirements on hardware…Mingfei Ma (Intel), Vitaly Fedyunin (Meta), Wei Wei (Meta)August 16, 2022
Introducing the PlayTorch app: Rapidly Create Mobile AI Experiences Blog Introducing the PlayTorch app: Rapidly Create Mobile AI Experiences In December, we announced PyTorch Live, a toolkit for building AI-powered mobile prototypes in minutes.…PlayTorch TeamJuly 22, 2022
What Every User Should Know About Mixed Precision Training in PyTorch Blog What Every User Should Know About Mixed Precision Training in PyTorch Efficient training of modern neural networks often relies on using lower precision data types. Peak…Syed Ahmed, Christian Sarofeen, Mike Ruberry, Eddie Yan, Natalia Gimelshein, Michael Carilli, Szymon Migacz, Piotr Bialecki, Paulius Micikevicius, Dusan Stosic, Dong Yang, and Naoya MaruyamaJuly 19, 2022