hydrax

The main Hydrax module, providing Dataloader, DataGroup, and Batch.

hydrax.image	Provides colorspace-aware utilities for loading and processing images with Pillow (PIL).
hydrax.pandas	Provides support for using Pandas DataFrames in a hydrax.DataGroup.
hydrax.tqdm	Provides tqdm progress bar support.

Batch	An abstract base class representing a batch of data loaded by a Dataloader.
DataGroup	Represents a group of data which share the same descriptor, batch size, and array shapes.
Dataloader	A zero-copy multiprocess JAX dataloader.
TrainingBatch	A batch of training data loaded by a Dataloader.
ValidationBatch	A batch of validation data loaded by a Dataloader.

debug_batch_references	Prints a partial tree of all traceable references to batch memory.
is_worker	Returns True if running in a subprocess.

BatchTimeoutError

Raised if no batch has been produced for some time.