modules

Modules

modelopt.torch.nas.modules.container	Dynamic sequential implementation (variable depth based on torch.nn.modules.container).
modelopt.torch.nas.modules.conv	Dynamic Conv implementations based on torch.nn.modules.conv.
modelopt.torch.nas.modules.linear	Dynamic linear implementations based on torch.nn.modules.linear.
modelopt.torch.nas.modules.norm	Dynamic norm implementations based on norm modules in torch.nn.modules.
modelopt.torch.nas.modules.utils	Internal module for utility functions.

The dynamic equivalents for torch.nn modules.