trace

Modules

modelopt.torch.trace.analyzer

Analyzer to look for layer-wise dependencies to identify connected symbols.

modelopt.torch.trace.modules

Module containing layer-specific symbolic information and implementations.

modelopt.torch.trace.plugins

Handles tracing plugins for third-party modules.

modelopt.torch.trace.symbols

Utilities to describe symbols found in common torch modules.

modelopt.torch.trace.tracer

Tracing module for building a set of partial Graphs describing the model.

Tracing module for layer-wise dependency analysis.