">np</span><span class="o">.</span><span class="n">ndarray</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="n">np</span><span class="o">.</span><span class="n">ndarray</span><span class="p">:</span>
    <span class="sd">&quot;&quot;&quot;</span>
<span class="sd">    Same as np.mean(array, axis=0) but njitted</span>
<span class="sd">    &quot;&quot;&quot;</span>
    <span class="n">mean_array</span> <span class="o">=</span> <span class="n">np</span><span class="o">.</span><span class=