Pytorch-like dataloaders in JAX.
deep-learning
tensorflow
pytorch
dataset
dataloader
datasets
jax
huggingface-datasets
jax-dataloader
-
Updated
Apr 2, 2024 - Jupyter Notebook
Pytorch-like dataloaders in JAX.
Add a description, image, and links to the jax-dataloader topic page so that developers can more easily learn about it.
To associate your repository with the jax-dataloader topic, visit your repo's landing page and select "manage topics."