DanLing NestedTensor: Composable Multi-Ragged Tensors for Deep Learning
This work presents DanLing NestedTensor, a PyTorch tensor abstraction that makes multi-ragged structure a property of the tensor itself, and lets model code built from its supported operators compose efficient variable-size computation without managing offsets at any call site.