Datapipe optimisation

@knoriy , mind taking a look? It looks similar to this one → Slow aws S3 data loading using TorchData open_files_by_fsspec