Skip to content

NexuML

tensor_shards_backend

`nexuml.data.loaders.tensor_shards_backend`¶

nexuml.data.loaders.tensor_shards_backend ¶

Windowed loader backend for tensor-shard exports.

TensorShardWindowLoader ¶

Bases: DataLoader

PyTorch DataLoader that receives already-batched shard-window items.

TensorShardsLoaderBackend ¶

Create split-specific loaders for tensor-shard exports.