Blog

Blog

Empowering Models with Performance: The Art of Generalized Model Transformation Approach

Introduction PyTorch 2.0 (PT2) offers a compiled execution mode which rewrites Python bytecode to extract sequences…

Jackie (Jiaqi) Xu, Yanbo Liang, Jason Ansel, Chunzhi Yang, Jade Nie, Yuzhen Huang, CK Luk, Xiaodong Wang, Lu Fang, Menglu Yu, Jinwon Lee, Daohang Shi, Flavio Sales TruzziDecember 15, 2023

Blog

Understanding GPU Memory 1: Visualizing All Allocations over Time

During your time with PyTorch on GPUs, you may be familiar with this common error…

Aaron Shi, Zachary DeVitoDecember 14, 2023

Blog

From PyTorch Conference 2023: From Dinosaurs to Seismic Imaging with Intel

Lightning Talk 1: Seismic Data to Subsurface Models with OpenFWI Speaker: Benjamin Consolvo, AI Software…

Ramya Ravi, Susan Kahler at IntelDecember 12, 2023

Community

PyPose: A Library for Robot Learning with Physics-based Optimization

We are excited to share our new open-source library PyPose. It is a PyTorch-based robotics-oriented…

PyPoseDecember 6, 2023

Announcements

Snowflake Joins the PyTorch Foundation as a General Member

The PyTorch Foundation, a neutral home for the deep learning community to collaborate on the…

PyTorch FoundationDecember 5, 2023

Blog

Accelerating Generative AI with PyTorch II: GPT, Fast

This post is the second part of a multi-series blog focused on how to accelerate…

PyTorch FoundationNovember 30, 2023

Blog

PyTorch 2.1 Contains New Performance Features for AI Developers

We are excited to see the release of PyTorch 2.1. In this blog, we discuss…

IntelNovember 29, 2023

Announcements

Lightning AI Joins the PyTorch Foundation as a Premier Member

The PyTorch Foundation, a neutral home for the deep learning community to collaborate on the…

PyTorch FoundationNovember 17, 2023

Announcements

🎉 PyTorch Docathon H2 2023 Wrap-up 🎉

We are thrilled to announce the successful completion of the Fall 2023 PyTorch Docathon! The…

PyTorch FoundationNovember 16, 2023

Blog

Accelerating Generative AI with PyTorch: Segment Anything, Fast

This post is the first part of a multi-series blog focused on how to accelerate…

PyTorch FoundationNovember 16, 2023

Community

How Activation Checkpointing enables scaling up training deep learning models

Activation checkpointing is a technique used for reducing the memory footprint at the cost of…

PyTorch FoundationNovember 9, 2023

Blog

PyTorch compile to speed up inference on Llama 2

In this blog, we discuss how to improve the inference latencies of the Llama 2…

IBM Research: Antoni Viros i Martin, Brian Vaughan, Davis Wertheimer, Joshua Rosenkranz, Mudhakar Srivatsa, Nelson Mimura Gonzalez, Raghu Ganti, Supriyo Chakraborty, Zhuoran Liu Meta: Geeta Chauhan, Hamid ShojanazeriNovember 7, 2023

Blog

High-Performance Llama 2 Training and Inference with PyTorch/XLA on Cloud TPUs

In a landscape where AI innovation is accelerating at an unprecedented pace, Meta’s Llama family of open…

Jiewen Tan, Jon Bolin, Yeounoh Chung, Liyang Lu, Siyuan Liu, Wonjoo Lee, Manfei Bai, Meghan Cowan, Jack Cao, Milad Mohammadi, Shauheen Zahirazami, Alex SpiridonovNovember 6, 2023

Blog

Accelerating Inference on x86-64 Machines with oneDNN Graph

Supported in PyTorch 2.0 as a beta feature, oneDNN Graph leverages aggressive fusion patterns to…

IntelNovember 2, 2023

Blog

AMD Extends Support for PyTorch Machine Learning Development on Select RDNA™ 3 GPUs with ROCm™ 5.7

Researchers and developers working with Machine Learning (ML) models and algorithms using PyTorch can now…

AMDOctober 31, 2023

Community

torch.compile, explained

Have you ever felt overwhelmed by the complexities of torch.compile? Diving into its workings can…

Kaichao YouOctober 26, 2023

Blog

Compiling NumPy code into C++ or CUDA via torch.compile

Quansight engineers have implemented support for tracing through NumPy code via torch.compile in PyTorch 2.1. This feature…

Evgeni Burovski, Ralf Gommers and Mario LezcanoOctober 17, 2023

Announcements

Huawei Joins the PyTorch Foundation as a Premier Member

Today, the PyTorch Foundation, a neutral home for the deep learning community to collaborate on…

PyTorch FoundationOctober 17, 2023

Announcements

PyTorch Edge: Enabling On-Device Inference Across Mobile and Edge Devices with ExecuTorch

We are excited to announce ExecuTorch, our all-new solution for enabling on-device inference capabilities across…

the PyTorch Edge TeamOctober 17, 2023

Blog

Flash-Decoding for long-context inference

Motivation Large language models (LLM) such as ChatGPT or Llama have received unprecedented attention lately.…

Tri Dao, Daniel Haziza, Francisco Massa, Grigory SizovOctober 13, 2023

Empowering Models with Performance: The Art of Generalized Model Transformation Approach

Understanding GPU Memory 1: Visualizing All Allocations over Time

From PyTorch Conference 2023: From Dinosaurs to Seismic Imaging with Intel

PyPose: A Library for Robot Learning with Physics-based Optimization

Snowflake Joins the PyTorch Foundation as a General Member

Accelerating Generative AI with PyTorch II: GPT, Fast

PyTorch 2.1 Contains New Performance Features for AI Developers

Lightning AI Joins the PyTorch Foundation as a Premier Member

🎉 PyTorch Docathon H2 2023 Wrap-up 🎉

Accelerating Generative AI with PyTorch: Segment Anything, Fast

How Activation Checkpointing enables scaling up training deep learning models

PyTorch compile to speed up inference on Llama 2

High-Performance Llama 2 Training and Inference with PyTorch/XLA on Cloud TPUs

Accelerating Inference on x86-64 Machines with oneDNN Graph

AMD Extends Support for PyTorch Machine Learning Development on Select RDNA™ 3 GPUs with ROCm™ 5.7

torch.compile, explained

Compiling NumPy code into C++ or CUDA via torch.compile

Huawei Joins the PyTorch Foundation as a Premier Member

PyTorch Edge: Enabling On-Device Inference Across Mobile and Edge Devices with ExecuTorch

Flash-Decoding for long-context inference

Docs

Tutorials

Resources

Stay in touch for updates, event info, and the latest news