diffalign.models.encoder.cross_attention

Classes

CrossAttention(dim[, heads, dropout, ...])	Q <- R masked dense cross-attention with optional coordinate updates.
CrossGraphAligner(dim[, heads, dropout, ...])	Masked dense cross-attention aligner (drop-in).