DataLoader

Auto-generated from jax_code_defs

PyTorch

API: torch.utils.data.DataLoader
Strategy: Direct Mapping
Official Docs ↗

JAX (Core)

API: GenericDataLoader
Strategy: Plugin (convert_dataloader)

TensorFlow

API: tf.data.Dataset
Strategy: Plugin (tf_data_loader)