ESM3 structure encoder for protein complexes

[williamberman]

Hello! Was the ESM3 structure encoder trained with all chains in a protein complex being jointly encoded or separately encoded (jointly meaning all coords for all chains in one complex passed as one batch element)? My assumption is separately as the geometric attention block uses the chain ids for the attn mask, but I'm not sure. Thank you!