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

device_mesh

device_mesh()

max.pipelines.lib.module_registration.device_mesh(devices, data_parallel_degree)

source

Returns the mesh registered modules are constructed on.

It is a dp axis under data parallelism, else a tp axis.

Parameters:

• devices (Sequence[Device])
• data_parallel_degree (int)

Return type:

DeviceMesh