PyTorch's banner

PyTorch

@PyTorch • 517,761 subscribers

Tensors and neural networks in Python with strong hardware acceleration. PyTorch is an open source project at the Linux Foundation. #PyTorchFoundation

Videos

Anya Rossi

sweetdream.ai

SweetDream.ai•Sponsored•Livecam

Watch Anya Live

Anya is streaming live right now! Join her private show and enjoy exclusive content.

Exclusive private shows

1.2k viewers online

Private Show

Join now for exclusive access

Free preview available • Premium content

How do you deliver full API parity across heterogeneous accelerators with zero additional overhead? At PyTorch Conference North America 2026, Zesheng Zong (Huawei), PyTorch Foundation Ambassador and Accelerator Integration Working Group Co-Chair, will share key technical updates on backend integration. The community is adapting over 600,000 upstream test cases to be generic across all backends, providing a scalable path toward complete API parity. Alongside this effort, Cross-Repository CI Relay connects every PyTorch pull request directly to accelerator repositories, enabling real-time bidirectional testing, early regression detection, and a lower barrier to entry for out-of-tree hardware maintainers. Join us in San Jose, California next week to see how these foundational updates simplify hardware enablement across the ecosystem: #PyTorchCon Zesheng

How do you deliver full API parity across heterogeneous accelerators with zero additional overhead? At PyTorch Conference North America 2026, Zesheng Zong (Huawei), PyTorch Foundation Ambassador and Accelerator Integration Working Group Co-Chair, will share key technical updates on backend integration. The community is adapting over 600,000 upstream test cases to be generic across all backends, providing a scalable path toward complete API parity. Alongside this effort, Cross-Repository CI Relay connects every PyTorch pull request directly to accelerator repositories, enabling real-time bidirectional testing, early regression detection, and a lower barrier to entry for out-of-tree hardware maintainers. Join us in San Jose, California next week to see how these foundational updates simplify hardware enablement across the ecosystem: #PyTorchCon Zesheng

14,135 просмотров • 6 дней назад

PyTorch Ambassador and #PyTorchCon North America 2026 program committee member Masahiro Hiramori (Masahiro Hiramori), 三菱電機｜Mitsubishi Electric, will be following work across on-device AI, efficient inference, voice agents, and machine learning compilers at this year’s conference. He's also looking forward to exchanging ideas with other developers and seeing what the broader PyTorch community is building. Register for PyTorch Conference North America 2026:

PyTorch Ambassador and #PyTorchCon North America 2026 program committee member Masahiro Hiramori (Masahiro Hiramori), 三菱電機｜Mitsubishi Electric, will be following work across on-device AI, efficient inference, voice agents, and machine learning compilers at this year’s conference. He's also looking forward to exchanging ideas with other developers and seeing what the broader PyTorch community is building. Register for PyTorch Conference North America 2026:

19,887 просмотров • 12 дней назад

When you run a model over a long context, the KV cache gets huge, so you have to throw some of it away. There are competing rules for choosing what to drop: keep the most recent tokens, keep the ones that get the most attention, and so on. The problem is that every method ships its rule welded to its own kernel. So if one method beats another, you can't tell whether the rule was smarter or the code was just faster. At #PyTorchCon North America, Huanran Wang of AMD will present the poster: FlexBudget: Interchangeable KV Importance Signals under One Compiled FlexAttention Budget We hold the kernel and the memory budget fixed, and swap only the rule. Same code, same allowance, honest comparison. It's training-free, and it's all core PyTorch. Join us in San Jose, October 20-21 to learn more:

When you run a model over a long context, the KV cache gets huge, so you have to throw some of it away. There are competing rules for choosing what to drop: keep the most recent tokens, keep the ones that get the most attention, and so on. The problem is that every method ships its rule welded to its own kernel. So if one method beats another, you can't tell whether the rule was smarter or the code was just faster. At #PyTorchCon North America, Huanran Wang of AMD will present the poster: FlexBudget: Interchangeable KV Importance Signals under One Compiled FlexAttention Budget We hold the kernel and the memory budget fixed, and swap only the rule. Same code, same allowance, honest comparison. It's training-free, and it's all core PyTorch. Join us in San Jose, October 20-21 to learn more:

13,038 просмотров • 10 дней назад

SGLang-Diffusion is a high-performance serving framework for diffusion models, designed for both large-scale offline generation and latency-sensitive real-time inference. At PyTorch Conference North America 2026, Yihao Wang and Kevin Mi of RadixArk will present the design of SGLang-Diffusion and the PyTorch-based inference optimizations behind it, including efficient request scheduling, memory management, batching strategies, and execution-path optimizations for diffusion pipelines. Join us in San Jose, October 20-21: #PyTorchCon

SGLang-Diffusion is a high-performance serving framework for diffusion models, designed for both large-scale offline generation and latency-sensitive real-time inference. At PyTorch Conference North America 2026, Yihao Wang and Kevin Mi of RadixArk will present the design of SGLang-Diffusion and the PyTorch-based inference optimizations behind it, including efficient request scheduling, memory management, batching strategies, and execution-path optimizations for diffusion pipelines. Join us in San Jose, October 20-21: #PyTorchCon

17,380 просмотров • 13 дней назад

Get more out of your AI infrastructure. Carlos Costa, Distinguished Engineer at IBM Research and co-founder of the llm-d project will share some exciting news about llm-d: how the project changed the way the open source AI community scales inference at PyTorch Conference North America. A little over a year after the project was founded, llm-d is reaching a new level of maturity, delivering state-of-the-art performance for generative serving, while also enabling the capabilities needed for day two operations, including multi-tenant quality of service, zero downtime operations, and all of that without sacrificing performance in production. Join us in San Jose, October 20-21:

Get more out of your AI infrastructure. Carlos Costa, Distinguished Engineer at IBM Research and co-founder of the llm-d project will share some exciting news about llm-d: how the project changed the way the open source AI community scales inference at PyTorch Conference North America. A little over a year after the project was founded, llm-d is reaching a new level of maturity, delivering state-of-the-art performance for generative serving, while also enabling the capabilities needed for day two operations, including multi-tenant quality of service, zero downtime operations, and all of that without sacrificing performance in production. Join us in San Jose, October 20-21:

11,448 просмотров • 10 дней назад

Think DeepSpeed is only good for ZeRO and its usability, not training throughput? At #PyTorchCon North America 2026, Masahiro Tanaka will cover tensor, sequence, and expert parallelism in DeepSpeed and walk through benchmarks. His session, “DeepSpeed Is Not Just ZeRO: Tensor, Sequence, and Expert Parallelism with ZeRO-Level Usability,” will cover AutoTP, AutoSP, and AutoEP, how these capabilities can be combined with ZeRO, and where each parallelism strategy fits. Register for PyTorch Conference North America 2026:

Think DeepSpeed is only good for ZeRO and its usability, not training throughput? At #PyTorchCon North America 2026, Masahiro Tanaka will cover tensor, sequence, and expert parallelism in DeepSpeed and walk through benchmarks. His session, “DeepSpeed Is Not Just ZeRO: Tensor, Sequence, and Expert Parallelism with ZeRO-Level Usability,” will cover AutoTP, AutoSP, and AutoEP, how these capabilities can be combined with ZeRO, and where each parallelism strategy fits. Register for PyTorch Conference North America 2026:

15,911 просмотров • 16 дней назад

PyTorch Conference is where the open source Al community gathers to talk about what we are all building together. Priya Nagpurkar, VP of AI Native Systems at IBM Research will give a keynote at #PyTorchCon North America on IBM & Red Hat's contributions to the PyTorch ecosystem, including exciting developments on infrastructure for out-of-tree accelerators, enablement of IBM's Spyre Accelerator, and work with Ilm-d for distributed inference. Come to the talk, meet the community, and hear what's next in open source Al. Join us in San Jose, October 20-21:

PyTorch Conference is where the open source Al community gathers to talk about what we are all building together. Priya Nagpurkar, VP of AI Native Systems at IBM Research will give a keynote at #PyTorchCon North America on IBM & Red Hat's contributions to the PyTorch ecosystem, including exciting developments on infrastructure for out-of-tree accelerators, enablement of IBM's Spyre Accelerator, and work with Ilm-d for distributed inference. Come to the talk, meet the community, and hear what's next in open source Al. Join us in San Jose, October 20-21:

16,181 просмотров • 18 дней назад

ERRC: Entropy-Reinvested Residual Correction for Tensor-Parallel LLM Communication PyTorch Foundation Ambassador Abdulsalam Bande will present a poster on ERRC (Entropy-Reinvested Residual Correction for Tensor-Parallel LLM Communication) at PyTorch Conference North America 2026. ERRC focuses on optimizing GPU data exchange during LLM inference. By compressing inter-GPU communication and using the saved bandwidth to mitigate quantization errors, ERRC makes multi-GPU inference significantly more efficient. Register and join us to San Jose, October 20-21: #PyTorchCon Abdulsalam Bande

ERRC: Entropy-Reinvested Residual Correction for Tensor-Parallel LLM Communication PyTorch Foundation Ambassador Abdulsalam Bande will present a poster on ERRC (Entropy-Reinvested Residual Correction for Tensor-Parallel LLM Communication) at PyTorch Conference North America 2026. ERRC focuses on optimizing GPU data exchange during LLM inference. By compressing inter-GPU communication and using the saved bandwidth to mitigate quantization errors, ERRC makes multi-GPU inference significantly more efficient. Register and join us to San Jose, October 20-21: #PyTorchCon Abdulsalam Bande

16,886 просмотров • 20 дней назад

From torch.profiler to Hardware Cycles: A Practical Profiling Playbook AWS Trainium: designed from the ground up for next generation AI training and inference is now a standard PyTorch device. When you need to optimize, the tools are built in. Esha Lakhotia and Pinak Panigrahi of Amazon Web Services will walk you from torch.profiler capture to cycle-level hardware debugging, with AI-assisted analysis that suggests fixes at PyTorch Conference North America. 📅 Oct 21 | 12:20 PM | LL21ABC Join us in San Jose, October 20-21: #PyTorchCon

From torch.profiler to Hardware Cycles: A Practical Profiling Playbook AWS Trainium: designed from the ground up for next generation AI training and inference is now a standard PyTorch device. When you need to optimize, the tools are built in. Esha Lakhotia and Pinak Panigrahi of Amazon Web Services will walk you from torch.profiler capture to cycle-level hardware debugging, with AI-assisted analysis that suggests fixes at PyTorch Conference North America. 📅 Oct 21 | 12:20 PM | LL21ABC Join us in San Jose, October 20-21: #PyTorchCon

14,523 просмотров • 17 дней назад

Want to hear more about new vLLM and SGLang engines built on a new PyTorch-native TPU backend, TorchTPU? Find Qi Zhou of Google, Colin Taylor and Angela Yi of Meta at PyTorch Conference North America in San Jose where they will discuss how they focused on making the existing SGLang & vLLM infrastructure work natively on TPU, preserving the schedulers, batching systems, OpenAI-compatible APIs, and torch.compile workflows already familiar to GPU users. Because the serving engines remain upstream, new models and features can run on TPU without requiring separate TPU-specific implementations. Learn more about how it works across the stacks & what they learned running this in production, including compile time, device placement under tracing, KV cache layout, and cross-chip collectives. Register for #PyTorchCon NA now:

Want to hear more about new vLLM and SGLang engines built on a new PyTorch-native TPU backend, TorchTPU? Find Qi Zhou of Google, Colin Taylor and Angela Yi of Meta at PyTorch Conference North America in San Jose where they will discuss how they focused on making the existing SGLang & vLLM infrastructure work natively on TPU, preserving the schedulers, batching systems, OpenAI-compatible APIs, and torch.compile workflows already familiar to GPU users. Because the serving engines remain upstream, new models and features can run on TPU without requiring separate TPU-specific implementations. Learn more about how it works across the stacks & what they learned running this in production, including compile time, device placement under tracing, KV cache layout, and cross-chip collectives. Register for #PyTorchCon NA now:

16,806 просмотров • 21 дней назад

Involved in the AI ecosystem? There's something for everyone at PyTorch Conference North America. Steve Pousty of Red Hat explains why you should attend from the new intro track added this year to poster sessions, booths, and colocated events. Steve is giving a talk on the use of PyTorch and PyTorch-based libraries from the ecosystem to do remote sensing for satellite imagery. Join us in San Jose, October 20-21: #PyTorchCon

Involved in the AI ecosystem? There's something for everyone at PyTorch Conference North America. Steve Pousty of Red Hat explains why you should attend from the new intro track added this year to poster sessions, booths, and colocated events. Steve is giving a talk on the use of PyTorch and PyTorch-based libraries from the ecosystem to do remote sensing for satellite imagery. Join us in San Jose, October 20-21: #PyTorchCon

12,529 просмотров • 15 дней назад

High-Velocity GPU Kernel Authoring with CUTLASS Python Join Guray Ozen from NVIDIA at PyTorch Conference North America 2026, where he will present recent improvements to CUTLASS for Python. CUTLASS for Python is the technology stack used to write speed-of-light kernels for the latest GPU generations, powering today's most critical AI workloads. In his presentation, Guray will introduce several new productivity features for developers and agents, including CuTe extensions, CUTLASS primitives, a task scheduler, compiler diagnostics, and static checks to help catch errors before launch. Join us on October 20-21 in San Jose: #PyTorchCon

High-Velocity GPU Kernel Authoring with CUTLASS Python Join Guray Ozen from NVIDIA at PyTorch Conference North America 2026, where he will present recent improvements to CUTLASS for Python. CUTLASS for Python is the technology stack used to write speed-of-light kernels for the latest GPU generations, powering today's most critical AI workloads. In his presentation, Guray will introduce several new productivity features for developers and agents, including CuTe extensions, CUTLASS primitives, a task scheduler, compiler diagnostics, and static checks to help catch errors before launch. Join us on October 20-21 in San Jose: #PyTorchCon

11,242 просмотров • 14 дней назад

What You Cannot Profile, You Cannot Optimize: Learning to Read PyTorch Traces In his talk at PyTorch Conference North America 2026, Suvaditya Mukherjee from Hugging Face will share practical guidance on how to start using the PyTorch Profiler on your models to extract maximum performance efficiency. Whether you are aiming to identify performance bottlenecks or optimize model execution, this talk will help you get the most out of your hardware. Join the open source AI community in San Jose, October 20-21: #PyTorchCon Suvaditya Mukherjee

16,264 просмотров • 22 дней назад

Build a neural network from scratch, train it, and optimize it for real-world AI infrastructure during our PyTorch Associate Training the day before PyTorch Conference North America on October 19th. Your instructor, Faradawn Yang of NVIDIA AI explains how this training will give you hands on experience with the latest AI accelerators & prepare you for the PyTorch Certified Associate (PTCA) certification exam. #PyTorchCon

Build a neural network from scratch, train it, and optimize it for real-world AI infrastructure during our PyTorch Associate Training the day before PyTorch Conference North America on October 19th. Your instructor, Faradawn Yang of NVIDIA AI explains how this training will give you hands on experience with the latest AI accelerators & prepare you for the PyTorch Certified Associate (PTCA) certification exam. #PyTorchCon

17,476 просмотров • 24 дней назад

Interested in optimizing Mixture of Experts LLM inference on ARM CPUs? In his talk at PyTorch Conference North America, Maajid Khan from Fujitsu Research India will explore efficient MoE LLM inference using vLLM and OpenVINO, sharing practical strategies for running these models effectively on ARM architectures. Register and join us in San Jose, October 20-21: #PyTorchCon

Interested in optimizing Mixture of Experts LLM inference on ARM CPUs? In his talk at PyTorch Conference North America, Maajid Khan from Fujitsu Research India will explore efficient MoE LLM inference using vLLM and OpenVINO, sharing practical strategies for running these models effectively on ARM architectures. Register and join us in San Jose, October 20-21: #PyTorchCon

16,448 просмотров • 28 дней назад

New chips are shipping faster than ever before, but the ecosystem is being held back by having to rewrite and then re-debug the same code over & over again. Chris Lattner CEO and Co-founder at Modular and EVP of Advanced AI Software & Platforms at Qualcomm, will deliver a keynote at PyTorch Conference North America about an open software platform for heterogeneous compute powered by Mojo and MAX. PyTorch has always been the place where the best models come together, and now there's a way to get those models onto all kinds of hardware. If you're interested in Al and compute, join us at the PyTorch Conference in San Jose, CA. Register now: #PyTorchCon

New chips are shipping faster than ever before, but the ecosystem is being held back by having to rewrite and then re-debug the same code over & over again. Chris Lattner CEO and Co-founder at Modular and EVP of Advanced AI Software & Platforms at Qualcomm, will deliver a keynote at PyTorch Conference North America about an open software platform for heterogeneous compute powered by Mojo and MAX. PyTorch has always been the place where the best models come together, and now there's a way to get those models onto all kinds of hardware. If you're interested in Al and compute, join us at the PyTorch Conference in San Jose, CA. Register now: #PyTorchCon

20,838 просмотров • 1 месяц назад

At #PyTorchCon North America 2026, Liz Li and Shekhar of AMD will discuss their work enabling scalable MXFP8 pretraining on MI355X GPUs. Liz and Shekhar will cover optimization of key MXFP8 operators in TorchAO and how those kernels were integrated with TorchTitan to build an end-to-end upstream PyTorch training stack for large language models. Their session, “Scaling MXFP8 Pretraining on 1K+ AMD Instinct MI355X: TorchAO Kernels and TorchTitan Training,” will also cover Triton and FlyDSL implementations and lessons from numerical validation, convergence, and performance tuning. Join us in San Jose on October 20-21. Register today:

At #PyTorchCon North America 2026, Liz Li and Shekhar of AMD will discuss their work enabling scalable MXFP8 pretraining on MI355X GPUs. Liz and Shekhar will cover optimization of key MXFP8 operators in TorchAO and how those kernels were integrated with TorchTitan to build an end-to-end upstream PyTorch training stack for large language models. Their session, “Scaling MXFP8 Pretraining on 1K+ AMD Instinct MI355X: TorchAO Kernels and TorchTitan Training,” will also cover Triton and FlyDSL implementations and lessons from numerical validation, convergence, and performance tuning. Join us in San Jose on October 20-21. Register today:

16,137 просмотров • 29 дней назад

Learn how PyTorch can extend language modeling beyond classification to practical time-to-event prediction. Witold Czubala of UBS will share a PyTorch-based transformer model that predicts six-month attrition from longitudinal advisor–client email communications for wealth management during PyTorch Conference North America in San Jose, CA. Join us October 20-21: #PyTorchCon

Learn how PyTorch can extend language modeling beyond classification to practical time-to-event prediction. Witold Czubala of UBS will share a PyTorch-based transformer model that predicts six-month attrition from longitudinal advisor–client email communications for wealth management during PyTorch Conference North America in San Jose, CA. Join us October 20-21: #PyTorchCon

17,124 просмотров • 1 месяц назад

The open source AI community thrives on collaboration, choice, and open contribution. As Joseph Groenenboom from Red Hat highlights, empowering developers to freely contribute and build options is essential for growing the entire AI ecosystem. PyTorch Conference North America 2026 is where that community comes together in person. Join us to collaborate with open source pioneers, expand your deployment capabilities, and directly contribute to shaping the future of machine learning. Register for PyTorch Conference North America today: #PyTorchCon

The open source AI community thrives on collaboration, choice, and open contribution. As Joseph Groenenboom from Red Hat highlights, empowering developers to freely contribute and build options is essential for growing the entire AI ecosystem. PyTorch Conference North America 2026 is where that community comes together in person. Join us to collaborate with open source pioneers, expand your deployment capabilities, and directly contribute to shaping the future of machine learning. Register for PyTorch Conference North America today: #PyTorchCon

17,740 просмотров • 1 месяц назад

"It's really great to feel the energy and see all the collaboration that's happening, lots of conversations. We have people here who are just starting to learn about PyTorch, and we have people who are maintainers. It's super nice to see that energy and see how together we're going to be able to solve some of these very difficult problems for humanity." Ankit Patel, NVIDIA, at PyTorch Conference Europe 2026 Register today to join the open source AI community at PyTorch Conference North America in San Jose, October 20–21: #PyTorchCon

"It's really great to feel the energy and see all the collaboration that's happening, lots of conversations. We have people here who are just starting to learn about PyTorch, and we have people who are maintainers. It's super nice to see that energy and see how together we're going to be able to solve some of these very difficult problems for humanity." Ankit Patel, NVIDIA, at PyTorch Conference Europe 2026 Register today to join the open source AI community at PyTorch Conference North America in San Jose, October 20–21: #PyTorchCon

17,000 просмотров • 1 месяц назад