Straggler Mitigation On PyTorch DDP By Hierarchical SGD Blog Straggler Mitigation On PyTorch DDP By Hierarchical SGD PyTorch DDP has been widely adopted across the industry for distributed training, which by default runs…Yi Wang (Cruise AI), Rohan Varma (Meta AI)April 7, 2023
Celebrate PyTorch 2.0 with New Performance Features for AI Developers Blog Celebrate PyTorch 2.0 with New Performance Features for AI Developers Congratulations to the PyTorch Foundation for its release of PyTorch 2.0! In this blog, I discuss…IntelApril 7, 2023
Profiling PyTorch language models with octoml-profile Community Profiling PyTorch language models with octoml-profile The recent launch of PyTorch 2.0 makes it clear that the community is heavily investing…Team OctoApril 4, 2023
PyTorch & OpenXLA: The Path Forward Blog PyTorch & OpenXLA: The Path Forward As we celebrate the release of OpenXLA, PyTorch 2.0, and PyTorch/XLA 2.0, it’s worth taking a step back…Milad Mohammadi, Jack Cao, Shauheen Zahirazami, Joe Spisak, and Jiewen TanApril 3, 2023
Accelerated PyTorch 2 Transformers Blog Accelerated PyTorch 2 Transformers The PyTorch 2.0 release includes a new high-performance implementation of the PyTorch Transformer API with…Michael Gschwind, Driss Guessous, Christian PuhrschMarch 28, 2023
PyTorch 2.0 & XLA—The Latest Cutting Edge Features Blog PyTorch 2.0 & XLA—The Latest Cutting Edge Features Today, we are excited to share our latest work for PyTorch/XLA 2.0. The release of PyTorch 2.0 is…Jack Cao, Milad Mohammadi, Alex Wertheim, Yeounoh Chung, Joe Spisak, Will Cromar, Shauheen ZahirazamiMarch 22, 2023
Accelerated Diffusers with PyTorch 2.0 Blog Accelerated Diffusers with PyTorch 2.0 PyTorch 2.0 has just been released. Its flagship new feature is torch.compile(), a one-line code change…Pedro Cuenca, Patrick von Platen, Suraj Patil, Sayak PaulMarch 16, 2023
PyTorch 2.0: Our next generation release that is faster, more Pythonic and Dynamic as ever Blog PyTorch 2.0: Our next generation release that is faster, more Pythonic and Dynamic as ever We are excited to announce the release of PyTorch® 2.0 which we highlighted during the PyTorch Conference on 12/2/22!…PyTorch FoundationMarch 15, 2023
New Library Updates in PyTorch 2.0 Blog New Library Updates in PyTorch 2.0 Summary We are bringing a number of improvements to the current PyTorch libraries, alongside the PyTorch…PyTorch FoundationMarch 15, 2023
How FASHABLE achieves SoA realistic AI generated images using PyTorch and Azure Machine Learning Community How FASHABLE achieves SoA realistic AI generated images using PyTorch and Azure Machine Learning Fashable is a company born at XNFY Lab (a joint initiative with Microsoft). The company’s…Orlando Ribas FernandesFebruary 23, 2023
Democratizing AI with PyTorch Foundation and ROCm™ support for PyTorch Blog Democratizing AI with PyTorch Foundation and ROCm™ support for PyTorch Last year, Meta announced that PyTorch joined the Linux Foundation as a neutral home for growing the…AMDFebruary 14, 2023
Deprecation of CUDA 11.6 and Python 3.7 Support Blog Deprecation of CUDA 11.6 and Python 3.7 Support For the upcoming PyTorch 2.0 feature release (target March 2023), we will target CUDA 11.7…PyTorch FoundationFebruary 2, 2023
Latest Colossal-AI boasts novel automatic parallelism and offers savings up to 46x for Stable Diffusion 2 Community Latest Colossal-AI boasts novel automatic parallelism and offers savings up to 46x for Stable Diffusion 2 As a new PyTorch Ecosystem Partner, we at HPC-AI Tech look forward to working with…Yang YouJanuary 31, 2023
PyTorch Trace Analysis for the Masses Blog PyTorch Trace Analysis for the Masses We are excited to announce the public release of Holistic Trace Analysis (HTA), an open…Anupam Bhatnagar, Xizhou Feng, Brian Coutinho, Yifan Liu, Sung-Han Lin, Louis Feng, and Yuzhen HuangJanuary 9, 2023
Distributed training with PyTorch and Azure ML Community Distributed training with PyTorch and Azure ML Suppose you have a very large PyTorch model, and you’ve already tried many common tricks…Beatriz StollnitzJanuary 6, 2023
Compromised PyTorch-nightly dependency chain between December 25th and December 30th, 2022. Blog Compromised PyTorch-nightly dependency chain between December 25th and December 30th, 2022. If you installed PyTorch-nightly on Linux via pip between December 25, 2022 and December 30,…PyTorch FoundationDecember 31, 2022
Torchserve Performance Tuning, Animated Drawings Case-Study Blog Torchserve Performance Tuning, Animated Drawings Case-Study In this post we discuss performance tuning of Torchserve for serving your models in production.…Hamid Shojanazeri, Geeta Chauhan, Mark Saroufim, Jesse SmithDecember 28, 2022
Scaling Vision Model Training Platforms with PyTorch Blog Scaling Vision Model Training Platforms with PyTorch TL;DR: We demonstrate the use of PyTorch with FairScale’s FullyShardedDataParallel (FSDP) API in writing large…Vaibhav Aggarwal, Mannat Singh, Anjali Sridhar, Yanghao Li, Shoubhik Debnath, Ronghang Hu, Will Feng, Xinlei Chen, Tingting Markstrum, Diana Liskovich, Anupam Bhatnagar, Chay Ryali, Haoqi Fan, Tete Xiao, Min Xu, Rahul Iyer, Christoph Feichtenhofer, Ross Girshick, Piotr Dollar, Aaron Adcock, Wan-Yen Lo, CK LukDecember 22, 2022
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