autoregression

Utilities for autoregression.

prxteinmpnn.utils.autoregression

prxteinmpnn.utils.autoregression.generate_ar_mask(decoding_order)

Get the autoregressive mask for the given decoding order.

Parameters:

decoding_order (Int[Array, 'num_residues']) – The order in which atoms are decoded.

Return type:

Int[Array, 'num_residues num_atoms']

Returns:

An atom mask where each atom can only attend to itself and previous atoms.