bridge.data.megatron_mimo

MegatronMIMO multi-encoder data loading utilities.

Submodules

• bridge.data.megatron_mimo.hf_provider
• bridge.data.megatron_mimo.reorder_buffer
• bridge.data.megatron_mimo.loaders
• bridge.data.megatron_mimo.dataset
• bridge.data.megatron_mimo.canonical_sampler
• bridge.data.megatron_mimo.collate
• bridge.data.megatron_mimo.dp_utils
• bridge.data.megatron_mimo.base_provider
• bridge.data.megatron_mimo.mock_provider
• bridge.data.megatron_mimo.sequence_pack

Package Contents

Data

__all__

API

bridge.data.megatron_mimo.__all__

[‘MegatronMIMODataset’, ‘megatron_mimo_collate_fn’, ‘MegatronMIMODatasetProvider’, ‘HFMegatronMIMODa…