Pretty easy to do, we should expose `find` in https://github.com/pytorch/vision/blob/c786d1244e6ece181c7d1ddf58134de2f4175bc4/torchvision/prototype/datasets/__init__.py#L14 cc @pmeier @bjuncek