trace
trace
¶
Cached numerical analysis inside a pure trainer trace, without eager debugging.
AnalysisTrace(trace: Callable[..., Any], target: type[nn.Module], select: Callable[[list[str]], str | list[str]], analyze: Callable[..., Any])
¶
Discover abstractly once; execute capture and analysis in the same JIT.