TensorDict provides a specialized dictionary class for PyTorch tensors that supports indexing, shape operations, device casting, and batch manipulation. It also offers tensorclass, a dataclass decorator for tensors. The library enables grouped tensor operations where all contained tensors move together to devices or storage, and handles batch dimensions consistently across heterogeneous data structures commonly used in reinforcement learning and other ML workflows.