_sign_grad&#39;</span><span class="p">]</span>

<span class="k">if</span> <span class="n">brevitas</span><span class="o">.</span><span class="n">NATIVE_STE_BACKEND_LOADED</span><span class="p">:</span>
    <span class="n">fn_prefix</span> <span class="o">=</span> <span class="n">torch</span>
    <span class="n">script_flag</span> <span class="o">=</span> <span class="n">brevitas</span><span class="o">.</span><span class="n">jit</span><span class="o">.</span><span class="n">script</span>
<span class="k">else