an><span class="kn">import</span> <span class="n">Parameter</span>

<span class="kn">import</span><span class="w"> </span><span class="nn">brevitas</span>
<span class="kn">import</span><span class="w"> </span><span class="nn">brevitas.config</span><span class="w"> </span><span class="k">as</span><span class="w"> </span><span class="nn">config</span>
<span class="kn">from</span><span class="w"> </span><span class="nn">brevitas.core.function_wrapper</span><span class="w"> </span><span class="kn">import</span>