hayden-donnelly/mnist-webdataset-png

MNIST WebDataset PNG The MNIST dataset with samples stored as PNG images and compiled into the WebDataset format. DALI/JAX Example The following code shows how this dataset can be loaded into JAX arrays by DALI. from nvidia.dali import pipeline_def import nvidia.dali.fn as fn import nvidia.dali.types as types from nvidia.dali.plugin.jax import DALIGenericIterator from nvidia.dali.plugin.base_iterator import LastBatchPolicy def get_data_iterator(batch_size… See the full description on the dataset page: https://huggingface.co/datasets/hayden-donnelly/mnist-webdataset-png.