"compile_enabled": bool(config.compile.enabled),
"validate_placement": bool(getattr(config.model, "validate_placement", False)),
},
)
get_batch = config.dataloader.get_batch.build(
mesh_context=self.base.mesh,
device=self.base.device,
data_config=getattr(config.dataset, "data_config", {}),
pp_shared_data=bool(getattr(config.dataloader, "pp_shared_data", False)),