pan class="sd">        Parameters</span>
<span class="sd">        ----------</span>
<span class="sd">        invariants</span>
<span class="sd">            List of MultipleMomentInvariant objects</span>
<span class="sd">        protein_keys</span>
<span class="sd">            list of protein names = rows of the output embedding.</span>
<span class="sd">            if None, takes all keys in `invariants`</span>
<span class="sd">        model</span>
<span class="sd">            if given, uses this model to ma