Skip to content

Conversation

@akoumpa
Copy link
Contributor

@akoumpa akoumpa commented Oct 28, 2025

Support registering out-of-tree strategies with a decorator:

from nemo_automodel.components.distributed.parallelizer import ParallelizationStrategy, register_parallel_strategy

@register_parallel_strategy
class MyParallelStrategy(ParallelizationStrategy):
    def parallelize(
        self,
        model: nn.Module,
        ...
    ) -> nn.Module:
Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>
@copy-pr-bot
Copy link

copy-pr-bot bot commented Oct 28, 2025

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>
@akoumpa akoumpa changed the title add oot parallelism registry Oct 28, 2025
@akoumpa akoumpa changed the title feat: add oot parallelism registry Oct 29, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

2 participants