DataLoader ========== Auto-generated from jax_code_defs .. raw:: html

PyTorch

API: torch.utils.data.DataLoader
Strategy: Direct Mapping

JAX (Core)

API: GenericDataLoader
Strategy: Plugin (convert_dataloader)

TensorFlow

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