Tensor subclasses allow you to add extend PyTorch with new types of tensors without having to write any C++. They have been used to implement DTensor, FP8, Nested Jagged Tensor and Complex Tensor. Rec...Show More
🎉 Join the #1 community of podcast lovers and never miss a great podcast.
Sign up
Recommendations