Function create_loader 

pub fn create_loader(
    world_size: usize,
    rank: usize,
    total_samples: usize,
) -> Result<DistributedLoader>

Create a distributed loader for a dataset

Arguments

• world_size - Total number of workers
• rank - Rank of this worker
• total_samples - Total samples in the dataset

Returns

• Ok(DistributedLoader) - The loader
• Err(DatasetsError) - If creation fails