For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /get-started.md).

Python function

construct_module

construct_module()

max.pipelines.lib.module_registration.construct_module(module_cls, config, mesh)

source

Constructs a registered module as module_cls(config) on mesh.

The caller enters F.lazy(), so the module holds no weights until it compiles.

Parameters:

• module_cls (type[Module[..., Any]])
• config (object)
• mesh (DeviceMesh)

Return type:

Module[…, Any]