Tutorials

🧠 Introduction: What Is torch.func? In PyTorch, most models are built using stateful objects like nn.Module. While this is intuitive and powerful, functional programming can unlock more control, composability, and…

🧠 Introduction: What Is torch.distributions? Probabilistic modeling is at the core of many machine learning and deep learning algorithms—from variational autoencoders (VAEs) to Bayesian inference. PyTorch offers a powerful, flexible…

🧠 Introduction: What Is torch.distributed.checkpoint? In large-scale distributed training, saving and restoring model state is not as simple as calling torch.save() and torch.load(). When training across many GPUs or nodes,…

🧠 Introduction: What Is torch.distributed.pipelining? As deep learning models become increasingly large, training them on a single GPU or even with traditional data parallelism becomes challenging. That's where pipeline parallelism…

🚀 Introduction: What Is torch.distributed.optim? In distributed deep learning, syncing model weights across devices is crucial for consistent training. That’s where torch.distributed.optim comes in. torch.distributed.optim is a PyTorch module that…

Introduction: What Is torch.distributed.tensor.parallel? torch.distributed.tensor.parallel is a module in PyTorch that provides tools to implement tensor parallelism—a technique used to split large model tensors (e.g., weights) across multiple GPUs. Unlike…

What is torch.distributed.fsdp.fully_shard? The fully_shard function is PyTorch's granular, module-level API for applying Fully Sharded Data Parallelism (FSDP) to specific model components. Unlike wrapping entire models with FSDP, fully_shard enables: Selective sharding of individual model components Mixed…

What is torch.distributed.fsdp? torch.distributed.fsdp (Fully Sharded Data Parallel) is PyTorch's advanced distributed training strategy that optimizes memory usage by sharding model parameters, gradients, and optimizer states across multiple GPUs. Unlike traditional DDP (DistributedDataParallel)…

What is torch.distributed.elastic? torch.distributed.elastic is PyTorch's framework for fault-tolerant, elastic distributed training that automatically adapts to cluster changes. Unlike static distributed training, elastic training: Handles node failures gracefully - Automatically recovers from worker crashes…