Distributed data loading with Petastore for distributed training