nexuml.data.loaders.definitions

nexuml.data.loaders.definitions

Typed definitions for built-in loader backends.

TorchLoader

Bases: LoaderBackendDefinition

Use the standard PyTorch data loader backend.

DaliLoader

Bases: LoaderBackendDefinition

Use the NVIDIA DALI data loader backend.

TensorShardsLoader

Bases: LoaderBackendDefinition

Configure the windowed tensor-shard loader backend.