Efficient Large-Scale Training with Pytorch FSDP and AWS Blog Efficient Large-Scale Training with Pytorch FSDP and AWS Cutting-edge AI models are becoming extremely large. The cost and overhead of training these models…Less Wright, Hamid Shojanazeri, Geeta ChauhanDecember 16, 2022
Scaling PyTorch FSDP for Training Foundation Models on IBM Cloud Blog Scaling PyTorch FSDP for Training Foundation Models on IBM Cloud Large model training using a cloud native approach is of growing interest for many enterprises…Linsong Chu, Less Wright, Hamid Shojanazeri, Sophia Wen, Raghu Ganti, Geeta ChauhanDecember 15, 2022
Get Started with PyTorch 2.0 Summary and Overview Blog Get Started with PyTorch 2.0 Summary and Overview Introducing PyTorch 2.0, our first steps toward the next generation 2-series release of PyTorch. Over…PyTorch FoundationDecember 2, 2022
Accelerating Hugging Face and TIMM models with PyTorch 2.0 Blog Accelerating Hugging Face and TIMM models with PyTorch 2.0 torch.compile() makes it easy to experiment with different compiler backends to make PyTorch code faster with…Mark SaroufimDecember 2, 2022
How PyTorch is bringing the power of AI to computers and smartphones Case Studies How PyTorch is bringing the power of AI to computers and smartphones Many of the experiences people enjoy on Facebook and Instagram are powered by artificial intelligence…PyTorch FoundationDecember 2, 2022
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
IBM Research: Bringing massive AI models to any cloud Case Studies IBM Research: Bringing massive AI models to any cloud The field of AI is in the middle of a revolution. In recent years, AI…PyTorch FoundationNovember 17, 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
Run inference at scale for OpenFold, a PyTorch-based protein folding ML model, using Amazon EKS Case Studies Run inference at scale for OpenFold, a PyTorch-based protein folding ML model, using Amazon EKS In drug discovery, understanding the 3D structure of proteins is key to assessing the ability…PyTorch FoundationOctober 25, 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
Optimize Protein Folding Costs with OpenFold on AWS Batch Case Studies Optimize Protein Folding Costs with OpenFold on AWS Batch Knowing the physical structure of proteins is an important part of the drug discovery process.…PyTorch FoundationOctober 4, 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