diffalign.models.encoder.egnn ============================= .. automodule:: diffalign.models.encoder.egnn .. rubric:: Functions .. autosummary:: unsorted_segment_sum .. rubric:: Classes .. autosummary:: EGNN EGNN_old E_GCL GNN MLP TransformerNN