lass="n">TensorReplacer</span>
<span class="kn">from</span> <span class="nn">parallelformers.policies.base</span> <span class="kn">import</span> <span class="n">Policy</span>
<span class="kn">from</span> <span class="nn">parallelformers.utils</span> <span class="kn">import</span> <span class="n">rsetattr</span>


<div class="viewcode-block" id="ParallelEngine"><a class="viewcode-back" href="../../../parallel/Engine.html#parallelformers.parallel.engine.ParallelEngine">[docs]</a><span class="k">class</span> <