Basic PyTorch tensor internals, including sizes, strides, offsets, and memory management, with practical examples for efficient deep learning workflows.
-
Updated
Nov 24, 2024 - Jupyter Notebook
Basic PyTorch tensor internals, including sizes, strides, offsets, and memory management, with practical examples for efficient deep learning workflows.
My upstream C++/OpenMP contributions to PyTorch Internals (Inductor BMM kernels, Autograd, & Compiler correctness).
Add a description, image, and links to the pytorch-internals topic page so that developers can more easily learn about it.
To associate your repository with the pytorch-internals topic, visit your repo's landing page and select "manage topics."