subblock_stats

Modules

`modelopt.torch.nas.subblock_stats.calc_runtime_stats`
`modelopt.torch.nas.subblock_stats.calc_subblock_params_and_memory`	Calculate memory usage and parameter counts for neural network subblocks.