Megatron's multi-modal data loader - View it on GitHub