I see there is a BufferedShuffleDataset in the source code: Add ShuffleDataset with buffer (#45290) · pytorch/pytorch@96540e9 · GitHub
However, I cannot find this back in the documentation and it seems not to be in release 1.7.1. When does this become available in the standard torch pip package?