variable-size-inputs

Tag

Cards List
#variable-size-inputs

DanLing NestedTensor: Composable Multi-Ragged Tensors for Deep Learning

arXiv cs.LG ↗ · 2d ago Cached

DanLing NestedTensor presents a PyTorch tensor abstraction for composable multi-ragged tensors, reducing padding waste in variable-size deep learning inputs and achieving significant speedups (e.g., 2.74×–3.39×) and memory efficiency improvements on A100 GPUs.

0 favorites 0 likes
← Back to home

Submit Feedback