Please consider adding TorchData (GitHub - pytorch/data: A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.) to the pyTorch container. It is a very useful and relatively new library that allows streaming of data to pyTorch. It is a small package but the used version needs to be aligned with that of pyTorch. It would, therefore, be nice if it could be included in the container.