JAX DataLoader

Contents:

  • Installation Guide
  • Usage Guide
  • API Reference
  • Examples
  • Changelog
JAX DataLoader
  • Index

Index

_ | A | B | C | D | M | N | P | S

_

  • __init__() (jax_dataloader.DataLoaderConfig method), [1]

A

  • augmentation (jax_dataloader.DataLoaderConfig attribute)
  • auto_batch_size (jax_dataloader.DataLoaderConfig attribute)

B

  • batch_size (jax_dataloader.DataLoaderConfig attribute)

C

  • cache_size (jax_dataloader.DataLoaderConfig attribute)

D

  • DataLoaderConfig (class in jax_dataloader)

M

  • multi_gpu (jax_dataloader.DataLoaderConfig attribute)

N

  • num_workers (jax_dataloader.DataLoaderConfig attribute)

P

  • pinned_memory (jax_dataloader.DataLoaderConfig attribute)
  • prefetch (jax_dataloader.DataLoaderConfig attribute)
  • progress_tracking (jax_dataloader.DataLoaderConfig attribute)

S

  • shuffle (jax_dataloader.DataLoaderConfig attribute)

© Copyright 2025, Kartikey Rawat.

Built with Sphinx using a theme provided by Read the Docs.