modules

Modules

modelopt.torch.trace.modules.concat

Module for supporting tracing of concat operations.

modelopt.torch.trace.modules.nn

Module containing native torch layer-specific symbolic information and implementations.

Module containing layer-specific symbolic information and implementations.