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
pipeline_stages
pipeline_stages()
max.pipelines.lib.pipeline_parallel.pipeline_stages(sizes)
Returns the stages of a partition, in order.
-
Parameters:
-
sizes (list[int]) – Each stage’s layer count, as
split_layers()returns. -
Return type: