      <div class="highlight"><pre><span class="lineno">48</span>    <span class="kn">from</span> <span class="nn">labml_nn.conv_mixer</span> <span class="kn">import</span> <span class="n">ConvMixerLayer</span><span class="p">,</span> <span class="n">ConvMixer</span><span class="p">,</span> <span class="n">ClassificationHead</span><span class="p">,</span> <span class="n">PatchEmbeddings</span></pre></div>
        </div>
    </div>
    <div class='section' id='section-9'>
        <div class='docs'>
          