Pytorch-like dataloaders in JAX.
deep-learning
tensorflow
pytorch
dataset
dataloader
datasets
jax
huggingface-datasets
jax-dataloader
-
Updated
Apr 2, 2024 - Jupyter Notebook