
PyTorch
@PyTorch • 471,028 subscribers
Tensors and neural networks in Python with strong hardware acceleration. PyTorch is an open source project at the Linux Foundation. #PyTorchFoundation
Videos

FlexAttention is a novel compiler-driven programming model that allows implementing the majority of attention variants in a few lines of idiomatic PyTorch code Boyuan Feng & Avik Chaudhuri show how many existing attention variants can be implemented via FlexAttention & that we achieve competitive performance compared to handwritten kernels. 📺 Watch the full video on YouTube:
PyTorch17,910 views • 11 months ago

Our next PyTorch Compiler Series video is live with Michael Lazos and Avik Chaudhuri discussing the key concepts behind CUTLASS and how to utilize torch.compile to customize primitives for a specific workload via autotuning & epilogue fusion 📺 Watch on YouTube:
PyTorch12,886 views • 11 months ago

Missed today’s live Q&A on PyTorch 2.6? The full recording is now available on our LinkedIn & YouTube. PyTorch Core Maintainer Nikita Shulga (Meta) answered key questions about the latest release, sharing his insights on CUDA, torch.compile, security improvements and more. 📺 Watch the full session here: In this clip, Nikita breaks down a crucial topic: 👉 What are the challenges of benchmarking to avoid performance regressions in PyTorch releases? #pytorch #machinelearning #opensource #ai
PyTorch11,052 views • 1 year ago
No more content to load