m</span><span class="w"> </span><span class="nn">brevitas.core.stats</span><span class="w"> </span><span class="kn">import</span> <span class="n">DEFAULT_MOMENTUM</span>
<span class="kn">from</span><span class="w"> </span><span class="nn">brevitas.core.utils</span><span class="w"> </span><span class="kn">import</span> <span class="n">ParameterWrapper</span>
<span class="kn">from</span><span class="w"> </span><span class="nn">brevitas.core.utils</span><span class="w"> </span><span class="kn">import</span> <s