Duplicated from  chaidiscovery/chai-1

Iambackup

/

chai-1

like 0

• "__torch__.torch.nn.modules.linear.___torch_mangle_650.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_132.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_738.Linear",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_246.TriangleAttentionUpdate_v2a",
• "__torch__.chai.model.primitives_fast.___torch_mangle_906.GatedSelfAttention",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_168.TriangleAttentionUpdate_v2a",
• "__torch__.torch.nn.modules.linear.___torch_mangle_632.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_731.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_338.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_967.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1279.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1102.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_850.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_604.Linear",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_674.PairformerBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1197.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_999.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_606.Linear",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_844.TriangleAttentionUpdate_v2a",
• "__torch__.torch.nn.modules.linear.___torch_mangle_535.Linear",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_1338.TriangleAttentionUpdate_v2a",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_309.AttentionPairBiasBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_468.Linear",
• "__torch__.chai.model.blocks3.transition_layer.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_291.Linear",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_380.TransitionLayer_v1",
• "__torch__.chai.model.primitives_fast.___torch_mangle_1270.GatedSelfAttention",
• "__torch__.chai.model.primitives_fast.___torch_mangle_542.GatedSelfAttention",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_441.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_242.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1263.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_797.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_111.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_889.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_845.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_133.Linear",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_22.TriangleAttentionUpdate_v2a",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1004.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1259.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_875.LayerNorm",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_860.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_785.Linear",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_1307.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_621.AttentionPairBiasBlock",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_615.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_166.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_209.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_981.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_339.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_295.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_160.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1227.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_791.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_625.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_425.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1080.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_872.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_879.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_0.Linear",
• "__torch__.torch.nn.modules.linear.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1311.Linear",
• "__torch__.chai.model.primitives_fast.___torch_mangle_464.GatedSelfAttention",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_948.TriangleAttentionUpdate_v2a",
• "__torch__.torch.nn.modules.linear.___torch_mangle_186.Linear",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_1078.TriangleAttentionUpdate_v2a",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1237.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_1009.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_5.Linear",
• "__torch__.chai.model.primitives_fast.___torch_mangle_1374.GatedSelfAttention",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_1011.AttentionPairBiasBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_223.Linear",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_995.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.torch.nn.modules.linear.___torch_mangle_404.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_832.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_407.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1331.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_567.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_684.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_781.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_138.Linear",
• "__torch__.chai.model.primitives_fast.___torch_mangle_620.GatedSelfAttention",
• "__torch__.einops.layers.torch.___torch_mangle_853.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1211.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_651.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_942.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_129.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1248.Linear",
• "__torch__.torch.nn.modules.activation.ReLU",
• "__torch__.torch.nn.modules.linear.___torch_mangle_534.Linear",
• "__torch__.chai.model.primitives_fast.___torch_mangle_438.GatedSelfAttention",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_720.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_586.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_148.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_410.EinMix",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_913.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_824.LayerNorm",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1016.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_864.Linear",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_1194.PairformerBlock",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_314.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_442.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_789.Linear",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_1312.TriangleAttentionUpdate_v2a",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_267.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1223.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_479.Linear",
• "__torch__.torch.nn.modules.container.Sequential",
• "__torch__.torch.nn.modules.linear.___torch_mangle_417.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_842.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_278.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1205.Linear",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_158.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_431.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_800.EinMix",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1146.TransitionLayer_v1",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_935.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1019.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_124.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_963.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1206.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_225.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1187.LayerNorm",
• "__torch__.einops.layers.torch.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_305.Linear",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_298.TriangleAttentionUpdate_v2a",
• "__torch__.torch.nn.modules.linear.___torch_mangle_227.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_746.LayerNorm",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_829.AttentionPairBiasBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_435.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_763.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_947.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_330.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1158.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_161.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_377.LayerNorm",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_1090.PairformerBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_989.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1124.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_798.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1285.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1293.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_884.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1007.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_398.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_229.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1384.Linear",
• "__torch__.chai.model.primitives_fast.___torch_mangle_1348.GatedSelfAttention",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_548.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_139.Linear",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_258.PairformerBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1358.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_601.LayerNorm",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_813.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_354.TransitionLayer_v1",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1317.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1091.LayerNorm",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_120.TriangleAttentionUpdate_v2a",
• "__torch__.torch.nn.modules.linear.___torch_mangle_112.Linear",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1198.TransitionLayer_v1",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1380.LayerNorm",
• "__torch__.chai.model.primitives_fast.___torch_mangle_1244.GatedSelfAttention",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1105.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_920.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1027.LayerNorm",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_1168.PairformerBlock",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_751.AttentionPairBiasBlock",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_402.TriangleAttentionUpdate_v2a",
• "__torch__.torch.nn.modules.linear.___torch_mangle_555.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1074.Linear",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_335.AttentionPairBiasBlock",
• "__torch__.einops.layers.torch.___torch_mangle_1294.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1362.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_571.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_477.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1201.Linear",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_205.AttentionPairBiasBlock",
• "__torch__.einops.layers.torch.___torch_mangle_1373.EinMix",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1030.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_665.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_14.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_1295.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_890.Linear",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_752.PairformerBlock",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_741.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_910.Linear",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_673.AttentionPairBiasBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_70.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_526.Linear",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_570.PairformerBlock",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_987.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1363.Linear",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_492.PairformerBlock",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_237.LayerNorm",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_67.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_843.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1367.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_1268.EinMix",
• "__torch__.chai.model.primitives_fast.___torch_mangle_1218.GatedSelfAttention",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1077.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1107.Linear",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_125.TriangleAttentionUpdate_v2a",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1175.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1337.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1051.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_658.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1097.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_550.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_384.EinMix",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_922.TriangleAttentionUpdate_v2a",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_637.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_783.LayerNorm",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_1379.Pairformer",
• "__torch__.torch.nn.modules.linear.___torch_mangle_137.Linear",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_596.PairformerBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1381.Linear",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_964.TransitionLayer_v1",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1084.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_507.LayerNorm",
• "__torch__.einops.layers.torch.___torch_mangle_593.EinMix",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1212.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_476.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_289.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_473.Linear",
• "__torch__.chai.model.primitives_fast.___torch_mangle_282.GatedSelfAttention",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_579.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1122.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_395.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_939.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1071.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_863.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_694.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_979.LayerNorm",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_709.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_336.PairformerBlock",
• "__torch__.einops.layers.torch.___torch_mangle_878.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_141.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_690.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_1008.EinMix",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_367.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_243.Linear",
• "__torch__.chai.model.primitives_fast.___torch_mangle_412.GatedSelfAttention",
• "__torch__.einops.layers.torch.___torch_mangle_670.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_968.Linear",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_210.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_270.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_946.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_679.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_707.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_618.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_222.Linear",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_371.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.torch.nn.modules.linear.___torch_mangle_424.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_304.LayerNorm",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_1271.AttentionPairBiasBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_729.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_150.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_164.Linear",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_986.PairformerBlock",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_787.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.einops.layers.torch.___torch_mangle_1060.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_712.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_274.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1254.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_799.Linear",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_865.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_341.LayerNorm",
• "__torch__.einops.layers.torch.___torch_mangle_176.EinMix",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_834.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_962.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_368.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_1139.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_394.Linear",
• "torch._utils._rebuild_tensor_v2",
• "__torch__.torch.nn.modules.linear.___torch_mangle_734.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_540.EinMix",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_100.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_426.Linear",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_163.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.torch.nn.modules.linear.___torch_mangle_65.Linear",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_302.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_866.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_871.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1315.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_355.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1291.LayerNorm",
• "__torch__.chai.model.blocks3.triangle_attention.TriangleAttentionUpdate_v2a",
• "__torch__.torch.nn.modules.linear.___torch_mangle_443.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_982.EinMix",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_719.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_949.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_171.Linear",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_848.TransitionLayer_v1",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_614.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_4.Linear",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_896.TriangleAttentionUpdate_v2a",
• "__torch__.torch.nn.modules.linear.___torch_mangle_92.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1153.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_664.Linear",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_756.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_578.Linear",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_1272.PairformerBlock",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_408.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1240.LayerNorm",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1328.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_342.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1176.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_575.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_251.LayerNorm",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_740.TriangleAttentionUpdate_v2a",
• "__torch__.torch.nn.modules.linear.___torch_mangle_888.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_582.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1013.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_456.Linear",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_662.TriangleAttentionUpdate_v2a",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_154.PairformerBlock",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_657.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.einops.layers.torch.___torch_mangle_722.EinMix",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_475.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_533.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_819.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_505.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_916.Linear",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_907.AttentionPairBiasBlock",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_1037.AttentionPairBiasBlock",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_94.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1135.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1173.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_758.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_966.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_635.Linear",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_778.PairformerBlock",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1239.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_136.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1335.Linear",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_792.TriangleAttentionUpdate_v2a",
• "__torch__.torch.nn.modules.linear.___torch_mangle_347.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1275.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_980.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_64.LayerNorm",
• "__torch__.chai.model.primitives_fast.___torch_mangle_672.GatedSelfAttention",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_484.TransitionLayer_v1",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_822.TransitionLayer_v1",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_259.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_649.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_841.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_333.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1345.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1277.LayerNorm",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_1349.AttentionPairBiasBlock",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1302.TransitionLayer_v1",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_917.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.torch.nn.modules.linear.___torch_mangle_972.Linear",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_574.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_820.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1159.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_327.Linear",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_1115.AttentionPairBiasBlock",
• "__torch__.chai.model.modules.af3_pairformer.PairformerBlock",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_262.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1067.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1029.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1253.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1303.LayerNorm",
• "__torch__.chai.model.af3.trunk.Trunk",
• "__torch__.chai.model.primitives_fast.___torch_mangle_1140.GatedSelfAttention",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_667.LayerNorm",
• "__torch__.einops.layers.torch.___torch_mangle_619.EinMix",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_49.LayerNorm",
• "__torch__.chai.model.blocks3.outer_product_mean.OPMv2c",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.torch.nn.modules.linear.___torch_mangle_502.Linear",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_206.PairformerBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_925.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_128.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_706.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_29.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_79.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_385.EinMix",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_1026.TriangleAttentionUpdate_v2a",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1174.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_903.Linear",
• "__torch__.chai.model.modules.af3_pair_weighted_averaging.MSAPairWeightedAveragingWithGating",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_990.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_244.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1044.Linear",
• "__torch__.chai.model.primitives_fast.___torch_mangle_360.GatedSelfAttention",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_428.TriangleAttentionUpdate_v2a",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1309.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_592.EinMix",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_1151.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.torch.nn.modules.linear.___torch_mangle_57.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1189.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_928.LayerNorm",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_439.AttentionPairBiasBlock",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_517.AttentionPairBiasBlock",
• "__torch__.chai.model.modules.af3_pairformer.Pairformer",
• "__torch__.einops.layers.torch.___torch_mangle_905.EinMix",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_1364.TriangleAttentionUpdate_v2a",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_1038.PairformerBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_446.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_563.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1109.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_531.Linear",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_189.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1214.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_157.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_101.Linear",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_17.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_146.LayerNorm",
• "__torch__.einops.layers.torch.___torch_mangle_488.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1028.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_422.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_170.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_597.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1225.LayerNorm",
• "__torch__.einops.layers.torch.___torch_mangle_1372.EinMix",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_714.TriangleAttentionUpdate_v2a",
• "__torch__.torch.nn.modules.linear.___torch_mangle_294.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_498.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_53.LayerNorm",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_803.AttentionPairBiasBlock",
• "__torch__.einops.layers.torch.___torch_mangle_904.EinMix",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_273.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1022.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1318.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_348.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_87.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1127.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_320.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_659.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1169.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1170.Linear",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_725.AttentionPairBiasBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_583.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_838.Linear",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_610.TriangleAttentionUpdate_v2a",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1371.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_460.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_287.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_656.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_159.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1041.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1351.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_82.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1252.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1360.Linear",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_284.PairformerBlock",
• "__torch__.chai.model.primitives_fast.___torch_mangle_1062.GatedSelfAttention",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_1359.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.torch.nn.modules.linear.___torch_mangle_847.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_1191.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_2.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_268.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_162.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_994.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_374.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_893.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_445.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_290.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_216.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_668.LayerNorm",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_959.AttentionPairBiasBlock",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_538.LayerNorm",
• "__torch__.chai.model.primitives_fast.___torch_mangle_334.GatedSelfAttention",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_455.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_292.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_240.Linear",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_184.TransitionLayer_v1",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_757.LayerNorm",
• "__torch__.einops.layers.torch.___torch_mangle_515.EinMix",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_849.LayerNorm",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_1297.AttentionPairBiasBlock",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_68.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1326.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_743.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_73.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_118.Linear",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_718.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_711.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1117.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_66.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1356.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_537.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_611.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1247.LayerNorm",
• "__torch__.chai.model.modules.af3_pair_weighted_averaging.___torch_mangle_52.MSAPairWeightedAveragingWithGating",
• "__torch__.torch.nn.modules.linear.___torch_mangle_821.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1131.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_721.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1210.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_547.Linear",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_440.PairformerBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1180.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_710.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_86.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1306.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_364.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1005.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_530.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_805.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_807.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_663.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_779.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_493.LayerNorm",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_194.TriangleAttentionUpdate_v2a",
• "__torch__.einops.layers.torch.___torch_mangle_775.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_301.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_401.Linear",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_466.PairformerBlock",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1250.TransitionLayer_v1",
• "__torch__.einops.layers.torch.___torch_mangle_1164.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_732.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_655.Linear",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_1073.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.torch.nn.modules.linear.___torch_mangle_143.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_497.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_914.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_370.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_644.EinMix",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_1255.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_870.TriangleAttentionUpdate_v2a",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1329.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1070.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1110.LayerNorm",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1342.TransitionLayer_v1",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_631.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1209.LayerNorm",
• "__torch__.einops.layers.torch.___torch_mangle_437.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_524.Linear",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_770.TransitionLayer_v1",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_324.TriangleAttentionUpdate_v2a",
• "__torch__.chai.model.primitives_fast.___torch_mangle_1296.GatedSelfAttention",
• "__torch__.torch.nn.modules.linear.___torch_mangle_399.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1100.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_11.Linear",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_84.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_175.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_772.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_836.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1157.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_937.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_851.Linear",
• "__torch__.chai.model.primitives_fast.___torch_mangle_490.GatedSelfAttention",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_231.AttentionPairBiasBlock",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_90.LayerNorm",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_692.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_788.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_823.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1001.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1015.Linear",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_1219.AttentionPairBiasBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1059.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_529.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_263.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1.LayerNorm",
• "__torch__.chai.model.primitives_fast.___torch_mangle_256.GatedSelfAttention",
• "__torch__.torch.nn.modules.linear.___torch_mangle_918.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_825.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_31.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_38.LayerNorm",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_636.TriangleAttentionUpdate_v2a",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_362.PairformerBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_996.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_461.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_174.LayerNorm",
• "__torch__.einops.layers.torch.___torch_mangle_281.EinMix",
• "__torch__.torch.nn.modules.container.___torch_mangle_59.ModuleList",
• "__torch__.torch.nn.modules.linear.___torch_mangle_50.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_416.Linear",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_856.PairformerBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1353.Linear",
• "__torch__.chai.model.primitives_fast.___torch_mangle_802.GatedSelfAttention",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_900.TransitionLayer_v1",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_1229.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.torch.nn.modules.linear.___torch_mangle_894.Linear",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_180.PairformerBlock",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_1298.PairformerBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_773.Linear",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_1167.AttentionPairBiasBlock",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_600.TransitionLayer_v1",
• "__torch__.einops.layers.torch.___torch_mangle_1061.EinMix",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_831.LayerNorm",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_640.TransitionLayer_v1",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1039.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_481.LayerNorm",
• "__torch__.einops.layers.torch.___torch_mangle_749.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_494.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_633.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_77.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_123.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_921.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_1320.EinMix",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_545.LayerNorm",
• "__torch__.einops.layers.torch.___torch_mangle_957.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1330.Linear",
• "__torch__.chai.model.primitives_fast.___torch_mangle_594.GatedSelfAttention",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_1234.TriangleAttentionUpdate_v2a",
• "__torch__.torch.nn.modules.linear.___torch_mangle_992.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_253.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1233.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_321.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1171.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_391.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1355.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_93.Linear",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_626.TransitionLayer_v1",
• "__torch__.chai.model.primitives_fast.___torch_mangle_1166.GatedSelfAttention",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1018.Linear",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_406.TransitionLayer_v1",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_874.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_708.Linear",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_1000.TriangleAttentionUpdate_v2a",
• "__torch__.torch.nn.modules.linear.___torch_mangle_25.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_641.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1236.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_639.Linear",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_622.PairformerBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1014.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_471.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_520.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_188.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_816.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1123.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_156.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1365.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_565.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_199.LayerNorm",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_432.TransitionLayer_v1",
• "__torch__.chai.model.primitives_fast.___torch_mangle_828.GatedSelfAttention",
• "__torch__.torch.nn.modules.linear.___torch_mangle_97.Linear",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_726.PairformerBlock",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_207.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_902.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_343.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_134.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_3.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_1113.EinMix",
• "__torch__.einops.layers.torch.___torch_mangle_1190.EinMix",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_419.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1057.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_883.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_409.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_973.Linear",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1108.TransitionLayer_v1",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_283.AttentionPairBiasBlock",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1273.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_747.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_1035.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_217.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_859.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_826.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1184.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_521.Linear",
• "__torch__.chai.model.primitives_fast.___torch_mangle_854.GatedSelfAttention",
• "__torch__.einops.layers.torch.___torch_mangle_1321.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_74.Linear",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_350.TriangleAttentionUpdate_v2a",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1316.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_472.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1336.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_627.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_786.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_113.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_689.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1049.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_358.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1257.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_487.Linear",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_215.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1304.Linear",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_648.PairformerBlock",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_1193.AttentionPairBiasBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_212.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_363.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_669.Linear",
• "__torch__.chai.model.primitives_fast.___torch_mangle_646.GatedSelfAttention",
• "__torch__.torch.nn.modules.linear.___torch_mangle_717.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_478.Linear",
• "__torch__.chai.model.modules.af3_msa_module.MSAModule",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_1182.TriangleAttentionUpdate_v2a",
• "__torch__.chai.model.primitives_fast.GatedSelfAttention",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1235.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_589.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_285.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_448.Linear",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_449.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1261.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_716.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1369.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_247.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_69.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1058.LayerNorm",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_1177.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.einops.layers.torch.___torch_mangle_332.EinMix",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_616.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_598.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_609.Linear",
• "__torch__.chai.model.primitives_fast.___torch_mangle_204.GatedSelfAttention",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1262.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_976.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_54.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_961.LayerNorm",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_796.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1149.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_469.Linear",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_506.TriangleAttentionUpdate_v2a",
• "__torch__.torch.nn.modules.linear.___torch_mangle_91.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_353.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_312.Linear",
• "__torch__.chai.model.blocks3.outer_product_mean.___torch_mangle_37.OPMv2c",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_13.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_383.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_705.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1069.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_643.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_202.EinMix",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_642.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_390.Linear",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_588.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_182.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_733.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_1112.EinMix",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_340.TransitionLayer_v1",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_397.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.torch.nn.modules.linear.___torch_mangle_48.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1341.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_983.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_15.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_485.LayerNorm",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_553.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_418.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_525.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_754.Linear",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_392.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_681.Linear",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1224.TransitionLayer_v1",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_595.AttentionPairBiasBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_617.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_654.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1207.Linear",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_1141.AttentionPairBiasBlock",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_532.TriangleAttentionUpdate_v2a",
• "__torch__.torch.nn.modules.linear.___torch_mangle_208.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1344.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_603.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_108.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_742.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_1138.EinMix",
• "__torch__.einops.layers.torch.___torch_mangle_827.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_685.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_628.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_806.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1289.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_951.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_462.EinMix",
• "__torch__.chai.model.primitives_fast.___torch_mangle_1322.GatedSelfAttention",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_699.AttentionPairBiasBlock",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_303.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_239.Linear",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_1323.AttentionPairBiasBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1352.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_352.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1033.Linear",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_288.TransitionLayer_v1",
• "__torch__.chai.model.primitives_fast.___torch_mangle_386.GatedSelfAttention",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_224.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_379.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1366.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_373.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_552.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_299.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_629.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_43.Linear",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_969.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.torch.nn.modules.linear.___torch_mangle_528.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_102.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_81.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_549.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_427.Linear",
• "__torch__.chai.model.primitives_fast.___torch_mangle_1010.GatedSelfAttention",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1017.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_233.LayerNorm",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_1012.PairformerBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_794.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1147.LayerNorm",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_366.TransitionLayer_v1",
• "__torch__.einops.layers.torch.___torch_mangle_931.EinMix",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_544.PairformerBlock",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1053.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_451.Linear",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_480.TriangleAttentionUpdate_v2a",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_881.AttentionPairBiasBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_815.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_103.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_452.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_988.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_323.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1178.Linear",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1056.TransitionLayer_v1",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1079.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_44.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_676.Linear",
• "__torch__.chai.model.primitives_fast.___torch_mangle_1088.GatedSelfAttention",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1048.Linear",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_496.TransitionLayer_v1",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_585.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_753.LayerNorm",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_257.AttentionPairBiasBlock",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_901.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1148.Linear",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_688.TriangleAttentionUpdate_v2a",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_777.AttentionPairBiasBlock",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.AttentionPairBiasBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_898.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1310.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_895.Linear",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_345.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1383.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_6.Linear",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_558.TriangleAttentionUpdate_v2a",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1096.Linear",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_1350.PairformerBlock",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_220.TriangleAttentionUpdate_v2a",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1305.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1319.Linear",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_179.AttentionPairBiasBlock",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_511.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_271.Linear",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_23.PairformerBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1327.Linear",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1042.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_728.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1308.Linear",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_647.AttentionPairBiasBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_572.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_680.Linear",
• "__torch__.chai.model.primitives_fast.___torch_mangle_308.GatedSelfAttention",
• "__torch__.torch.nn.modules.linear.___torch_mangle_869.Linear",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_387.AttentionPairBiasBlock",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_909.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1002.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_677.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_75.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_857.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_953.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1185.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_459.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_814.Linear",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1264.TransitionLayer_v1",
• "__torch__.einops.layers.torch.___torch_mangle_1217.EinMix",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1083.LayerNorm",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_293.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_311.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_372.Linear",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_130.TriangleAttentionUpdate_v2a",
• "__torch__.torch.nn.modules.linear.___torch_mangle_235.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1340.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1119.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_144.Linear",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_1286.TriangleAttentionUpdate_v2a",
• "__torch__.torch.nn.modules.linear.___torch_mangle_121.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1280.Linear",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_891.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.einops.layers.torch.___torch_mangle_1034.EinMix",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_467.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_277.LayerNorm",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_319.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.torch.nn.modules.linear.___torch_mangle_703.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1076.Linear",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_1047.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_1375.AttentionPairBiasBlock",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_735.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.chai.model.primitives_fast.___torch_mangle_1114.GatedSelfAttention",
• "__torch__.einops.layers.torch.___torch_mangle_280.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1357.Linear",
• "__torch__.chai.model.primitives_fast.___torch_mangle_984.GatedSelfAttention",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_55.LayerNorm",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1172.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_7.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_508.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_812.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1287.LayerNorm",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1160.TransitionLayer_v1",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1183.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_858.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_846.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_486.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_840.Linear",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_198.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_551.Linear",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_328.TransitionLayer_v1",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_173.LayerNorm",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_522.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_691.Linear",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_952.TransitionLayer_v1",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1213.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_877.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1284.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_602.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_62.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_95.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_351.LayerNorm",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_470.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_581.Linear",
• "__torch__.torch.nn.modules.container.___torch_mangle_72.ModuleList",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1278.Linear",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_1116.PairformerBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1129.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1258.Linear",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_276.TransitionLayer_v1",
• "__torch__.einops.layers.torch.___torch_mangle_541.EinMix",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1354.TransitionLayer_v1",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_30.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_415.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_56.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1032.LayerNorm",
• "__torch__.einops.layers.torch.___torch_mangle_307.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_970.Linear",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_804.PairformerBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_313.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1045.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1231.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_463.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1050.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1132.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_489.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_474.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_624.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_269.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_192.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_1269.EinMix",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_63.TransitionLayer_v1",
• "__torch__.chai.model.blocks3.outer_product_mean.___torch_mangle_34.OPMv2c",
• "__torch__.torch.nn.modules.linear.___torch_mangle_20.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1106.Linear",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_1125.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.torch.nn.modules.linear.___torch_mangle_369.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_768.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1222.Linear",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1134.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_503.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_977.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1195.LayerNorm",
• "__torch__.einops.layers.torch.___torch_mangle_645.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_556.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_697.EinMix",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1325.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_346.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_755.Linear",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_652.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_560.Linear",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_730.TransitionLayer_v1",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_897.LayerNorm",
• "__torch__.chai.model.primitives_fast.___torch_mangle_698.GatedSelfAttention",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_830.PairformerBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_51.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_696.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1081.Linear",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_666.TransitionLayer_v1",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1339.LayerNorm",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_908.PairformerBlock",
• "__torch__.einops.layers.torch.___torch_mangle_1165.EinMix",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_693.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_27.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1101.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_33.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1020.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1314.Linear",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_1142.PairformerBlock",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_252.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_576.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_930.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_780.Linear",
• "__torch__.chai.model.primitives_fast.___torch_mangle_880.GatedSelfAttention",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1313.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_695.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_873.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_924.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_331.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_195.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1066.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_495.Linear",
• "__torch__.torch.nn.modules.container.___torch_mangle_1385.Sequential",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_99.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_683.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1031.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_429.LayerNorm",
• "__torch__.einops.layers.torch.___torch_mangle_359.EinMix",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_42.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_661.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1024.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_356.LayerNorm",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1290.TransitionLayer_v1",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_76.TransitionLayer_v1",
• "__torch__.einops.layers.torch.___torch_mangle_1086.EinMix",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1343.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_941.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1118.Linear",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_518.PairformerBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_765.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_1216.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1228.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_61.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_950.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1282.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_35.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_105.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_940.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_264.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1054.Linear",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_562.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_318.Linear",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_491.AttentionPairBiasBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_682.Linear",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_1260.TriangleAttentionUpdate_v2a",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_675.LayerNorm",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_310.PairformerBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_971.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_83.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_745.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_923.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_349.Linear",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_912.TransitionLayer_v1",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_938.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_580.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_810.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_36.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_260.Linear",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_104.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_221.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1230.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1161.LayerNorm",
• "__torch__.torch.nn.modules.container.___torch_mangle_28.Sequential",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1145.Linear",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_700.PairformerBlock",
• "__torch__.einops.layers.torch.___torch_mangle_1087.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1098.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1103.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_116.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_403.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_24.LayerNorm",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_1208.TriangleAttentionUpdate_v2a",
• "__torch__.einops.layers.torch.___torch_mangle_514.EinMix",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_71.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_634.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_835.LayerNorm",
• "__torch__.chai.model.primitives_fast.___torch_mangle_1036.GatedSelfAttention",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_211.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_96.Linear",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_744.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_557.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_337.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1256.Linear",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_1089.AttentionPairBiasBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_122.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_767.LayerNorm",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_882.PairformerBlock",
• "__torch__.chai.model.modules.af3_pair_weighted_averaging.___torch_mangle_58.MSAPairWeightedAveragingWithGating",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_771.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_45.Linear",
• "__torch__.chai.model.modules.af3_pairformer.SquashNorm",
• "__torch__.chai.model.primitives_fast.___torch_mangle_230.GatedSelfAttention",
• "__torch__.torch.nn.modules.linear.___torch_mangle_868.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_127.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1181.Linear",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1120.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_191.Linear",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_135.TransitionLayer_v1",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_433.LayerNorm",
• "__torch__.einops.layers.torch.___torch_mangle_801.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1093.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_119.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_193.Linear",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_1246.PairformerBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_509.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_26.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_944.Linear",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_839.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_393.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_152.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_181.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_997.Linear",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_974.TriangleAttentionUpdate_v2a",
• "__torch__.einops.layers.torch.___torch_mangle_254.EinMix",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_382.LayerNorm",
• "__torch__.einops.layers.torch.___torch_mangle_203.EinMix",
• "__torch__.chai.model.primitives_fast.___torch_mangle_776.GatedSelfAttention",
• "__torch__.torch.nn.modules.linear.___torch_mangle_539.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_317.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_365.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_296.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1251.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_421.Linear",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_704.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_114.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_927.LayerNorm",
• "__torch__.chai.model.modules.af3_template_embedding.TemplateEmbedder",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1155.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1065.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_300.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1111.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_554.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_892.Linear",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_1021.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.einops.layers.torch.___torch_mangle_852.EinMix",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_1203.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_140.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_1063.AttentionPairBiasBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1215.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1144.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_790.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_447.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_482.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1232.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1226.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_748.EinMix",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_1156.TriangleAttentionUpdate_v2a",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1121.LayerNorm",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_109.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_569.AttentionPairBiasBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_769.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_306.EinMix",
• "__torch__.einops.layers.torch.___torch_mangle_411.EinMix",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_458.TransitionLayer_v1",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_12.TransitionLayer_v1",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_1220.PairformerBlock",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_818.TriangleAttentionUpdate_v2a",
• "__torch__.torch.nn.modules.linear.___torch_mangle_660.Linear",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_943.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.torch.nn.modules.linear.___torch_mangle_573.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_919.Linear",
• "__torch__.torch.nn.modules.normalization.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1006.LayerNorm",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_960.PairformerBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1196.Linear",
• "__torch__.torch.nn.modules.container.___torch_mangle_1377.ModuleList",
• "__torch__.torch.nn.modules.linear.___torch_mangle_219.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_107.Linear",
• "__torch__.torch.nn.modules.container.___torch_mangle_1382.Sequential",
• "__torch__.torch.nn.modules.linear.___torch_mangle_929.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_226.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_430.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1274.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_991.LayerNorm",
• "__torch__.einops.layers.torch.___torch_mangle_671.EinMix",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_172.TransitionLayer_v1",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1094.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1025.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_400.Linear",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_1245.AttentionPairBiasBlock",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_590.LayerNorm",
• "__torch__.einops.layers.torch.___torch_mangle_1347.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_190.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_389.LayerNorm",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_413.AttentionPairBiasBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_833.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_607.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1003.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_723.EinMix",
• "__torch__.einops.layers.torch.___torch_mangle_566.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_396.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_275.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_736.Linear",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_926.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1150.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_739.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_623.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_759.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_861.LayerNorm",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_465.AttentionPairBiasBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1075.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_234.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_998.Linear",
• "__torch__.torch.nn.modules.container.___torch_mangle_131.ModuleList",
• "__torch__.torch.nn.modules.linear.___torch_mangle_453.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_1346.EinMix",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_523.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1200.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_500.Linear",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_444.TransitionLayer_v1",
• "__torch__.einops.layers.torch.___torch_mangle_255.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_608.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1267.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1179.Linear",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_543.AttentionPairBiasBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_238.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_60.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_577.Linear",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_1099.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.torch.nn.modules.linear.___torch_mangle_201.Linear",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_510.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_248.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_686.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_899.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1043.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_185.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_450.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_638.Linear",
• "__torch__.torch.nn.modules.container.___torch_mangle_41.ModuleList",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1128.Linear",
• "__torch__.chai.model.primitives_fast.___torch_mangle_568.GatedSelfAttention",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_564.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1023.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1152.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1301.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_559.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_876.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_727.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1202.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_1242.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_867.Linear",
• "__torch__.chai.model.blocks3.outer_product_mean.___torch_mangle_40.OPMv2c",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_512.LayerNorm",
• "__torch__.chai.model.primitives_fast.___torch_mangle_932.GatedSelfAttention",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1186.TransitionLayer_v1",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_149.TransitionLayer_v1",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_361.AttentionPairBiasBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_513.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_169.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_405.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_956.EinMix",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1188.LayerNorm",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_1130.TriangleAttentionUpdate_v2a",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1299.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1046.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1292.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_329.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1163.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_457.Linear",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_527.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.torch.nn.modules.linear.___torch_mangle_117.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1095.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_261.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_737.Linear",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_1104.TriangleAttentionUpdate_v2a",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_1064.PairformerBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_183.Linear",
• "__torch__.chai.model.primitives_fast.___torch_mangle_1192.GatedSelfAttention",
• "__torch__.torch.nn.modules.linear.___torch_mangle_378.Linear",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_88.TransitionLayer_v1",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_145.TriangleAttentionUpdate_v2a",
• "__torch__.torch.nn.modules.linear.___torch_mangle_837.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_630.Linear",
• "__torch__.torch.nn.modules.container.___torch_mangle_89.ModuleList",
• "__torch__.torch.nn.modules.linear.___torch_mangle_591.Linear",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_423.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_808.TransitionLayer_v1",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_701.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1332.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_809.LayerNorm",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_985.AttentionPairBiasBlock",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_715.LayerNorm",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_250.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_213.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_687.Linear",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_886.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_420.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_245.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_126.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_885.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_147.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_325.LayerNorm",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1368.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_784.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_774.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_279.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_197.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_46.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_702.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1370.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_945.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_561.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_214.Linear",
• "__torch__.chai.model.primitives_fast.___torch_mangle_724.GatedSelfAttention",
• "__torch__.torch.nn.modules.linear.___torch_mangle_249.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_381.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_18.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_499.Linear",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_501.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.torch.nn.modules.linear.___torch_mangle_546.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_915.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1334.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_326.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1221.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1055.Linear",
• "__torch__.torch.nn.modules.container.___torch_mangle_110.ModuleList",
• "__torch__.torch.nn.modules.linear.___torch_mangle_936.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_795.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_47.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_375.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_155.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_975.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_286.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1241.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_599.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1265.LayerNorm",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_80.TransitionLayer_v1",
• "__torch__.einops.layers.torch.___torch_mangle_1243.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_764.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_196.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_98.Linear",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_1378.SquashNorm",
• "__torch__.torch.nn.modules.container.ModuleList",
• "__torch__.torch.nn.modules.linear.___torch_mangle_78.Linear",
• "__torch__.chai.model.primitives_fast.___torch_mangle_958.GatedSelfAttention",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_782.TransitionLayer_v1",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_414.PairformerBlock",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1266.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_151.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_887.LayerNorm",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_115.TriangleAttentionUpdate_v2a",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_584.TriangleAttentionUpdate_v2a",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_232.PairformerBlock",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_315.LayerNorm",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_376.TriangleAttentionUpdate_v2a",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_1052.TriangleAttentionUpdate_v2a",
• "__torch__.torch.nn.modules.linear.___torch_mangle_266.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1361.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1085.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1092.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_504.Linear",
• "__torch__.chai.model.primitives_fast.___torch_mangle_750.GatedSelfAttention",
• "__torch__.torch.nn.modules.linear.___torch_mangle_762.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1300.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_16.Linear",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_978.TransitionLayer_v1",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1068.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_187.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_322.Linear",
• "__torch__.chai.model.primitives_fast.___torch_mangle_516.GatedSelfAttention",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_1333.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_793.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_218.Linear",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_766.TriangleAttentionUpdate_v2a",
• "__torch__.torch.nn.modules.linear.___torch_mangle_713.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_653.LayerNorm",
• "torch.FloatStorage",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_388.PairformerBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_811.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_993.Linear",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1276.TransitionLayer_v1",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_200.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_817.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_862.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_142.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_344.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_228.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_955.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_106.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_177.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1072.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1249.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1136.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1137.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_357.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_434.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1288.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1199.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_10.Linear",
• "collections.OrderedDict",
• "__torch__.torch.nn.modules.linear.___torch_mangle_760.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_39.Linear",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_761.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_1324.PairformerBlock",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1082.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1154.Linear",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1238.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1126.Linear",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_678.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_19.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_911.Linear",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_934.PairformerBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1040.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_167.Linear",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_1376.PairformerBlock",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_536.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_21.Linear",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_241.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_454.TriangleAttentionUpdate_v2a",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_605.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.torch.nn.modules.linear.___torch_mangle_8.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_165.Linear",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_1281.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.torch.nn.modules.linear.___torch_mangle_613.Linear",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_236.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_612.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_587.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_153.EinMix",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_954.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1162.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1133.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_32.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_297.Linear",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_855.AttentionPairBiasBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1283.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_316.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1143.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_519.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_965.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1204.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_265.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_9.LayerNorm",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_272.TriangleAttentionUpdate_v2a",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_933.AttentionPairBiasBlock",
• "__torch__.chai.model.primitives_fast.___torch_mangle_178.GatedSelfAttention",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_85.LayerNorm",
• "__torch__.einops.layers.torch.___torch_mangle_436.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_483.Linear",
• "torch.LongStorage",
• "torch._utils._rebuild_tensor_v2",
• "collections.OrderedDict",
• "collections.OrderedDict",
• "torch.BoolStorage",
• "torch.BFloat16Storage",
• "torch.jit._pickle.restore_type_tag",
• "torch._utils._rebuild_tensor_v2"

How to fix it?