chaidiscovery

/

chai-1

like 14

Follow

Chai Discovery 5

• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1902.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1947.Linear",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_1984.AttentionPairBiasBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1937.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1971.Linear",
• "torch.FloatStorage",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1922.Linear",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_1948.TriangleAttentionUpdate_v1",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1924.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1961.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1966.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1918.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1938.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1945.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1898.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1943.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1974.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1894.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1940.LayerNorm",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_1919.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.einops.layers.torch.___torch_mangle_1956.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1967.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1942.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1887.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1899.LayerNorm",
• "__torch__.einops.layers.torch.___torch_mangle_1931.EinMix",
• "__torch__.torch.nn.modules.container.___torch_mangle_1986.ModuleList",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1897.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1911.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1965.LayerNorm",
• "__torch__.chai.model.blocks3.triangle_attention.TriangleAttentionUpdate_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1921.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1951.Linear",
• "__torch__.chai.model.modules.af3_confidence_head.ConfidenceHead",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_1960.PairformerBlock",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_1909.AttentionPairBiasBlock",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1949.LayerNorm",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1964.TransitionLayer_v1",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1928.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1970.LayerNorm",
• "collections.OrderedDict",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1963.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1916.Linear",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_1973.TriangleAttentionUpdate_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1962.Linear",
• "__torch__.chai.model.primitives_fast.___torch_mangle_1983.GatedSelfAttention",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1892.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_1957.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1885.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1915.LayerNorm",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_1985.PairformerBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1913.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1929.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1891.LayerNorm",
• "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_1923.TriangleAttentionUpdate_v1",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1952.TransitionLayer_v1",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1904.LayerNorm",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_1895.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1905.Linear",
• "__torch__.chai.model.primitives_fast.___torch_mangle_1908.GatedSelfAttention",
• "__torch__.einops.layers.torch.___torch_mangle_1932.EinMix",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1920.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1972.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1950.Linear",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_1959.AttentionPairBiasBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1976.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1896.LayerNorm",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1927.TransitionLayer_v1",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1978.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1941.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1975.Linear",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_1969.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1917.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1900.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1886.Linear",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_1934.AttentionPairBiasBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1925.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1926.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1946.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1989.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1901.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_1906.EinMix",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1914.TransitionLayer_v1",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1977.TransitionLayer_v1",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1890.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1889.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1912.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1988.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1953.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1903.LayerNorm",
• "__torch__.einops.layers.torch.___torch_mangle_1907.EinMix",
• "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_1944.TriangleMultiplicativeUpdateFastv2r",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_1910.PairformerBlock",
• "__torch__.chai.model.primitives_fast.___torch_mangle_1933.GatedSelfAttention",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1936.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1888.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1930.Linear",
• "torch._utils._rebuild_tensor_v2",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1954.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1955.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1987.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_1981.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1980.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1893.Linear",
• "__torch__.chai.model.primitives_fast.___torch_mangle_1958.GatedSelfAttention",
• "__torch__.einops.layers.torch.___torch_mangle_1982.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1968.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1979.LayerNorm",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1939.TransitionLayer_v1",
• "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_1935.PairformerBlock",
• "torch.LongStorage",
• "torch._utils._rebuild_tensor_v2",
• "collections.OrderedDict",
• "torch.FloatStorage",
• "collections.OrderedDict",
• "torch.LongStorage",
• "torch.BoolStorage",
• "torch.BFloat16Storage",
• "torch.jit._pickle.restore_type_tag",
• "torch._utils._rebuild_tensor_v2"

How to fix it?

• "__torch__.torch.nn.modules.linear.___torch_mangle_1845.Linear",
• "__torch__.torch.nn.modules.container.___torch_mangle_1834.ModuleList",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1656.Linear",
• "__torch__.torch.nn.modules.container.___torch_mangle_1524.Sequential",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1490.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1760.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1874.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1639.Linear",
• "__torch__.torch.nn.modules.container.___torch_mangle_1575.Sequential",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1728.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_1552.EinMix",
• "__torch__.torch.nn.modules.container.___torch_mangle_1521.Sequential",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1584.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1488.Linear",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1643.AdaLayerNorm_v2",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1859.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1686.LayerNorm",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1581.AdaLayerNorm_v2",
• "__torch__.torch.nn.modules.activation.___torch_mangle_1522.ReLU",
• "__torch__.chai.model.modules.af3_diffusion_transformer.___torch_mangle_1833.DiffusionTransformerBlock",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1622.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1735.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1677.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_1569.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1811.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1635.Linear",
• "__torch__.torch.nn.modules.container.___torch_mangle_1548.ModuleList",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1694.Linear",
• "__torch__.torch.nn.modules.container.___torch_mangle_1605.Sequential",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1770.Linear",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1777.AdaLayerNorm_v2",
• "__torch__.chai.model.blocks3.cond_transition.___torch_mangle_1599.ConditionedTransitionBlock_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1561.Linear",
• "__torch__.chai.model.modules.af3_diffusion_transformer.___torch_mangle_1609.DiffusionTransformerBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1835.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1623.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1616.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1624.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1494.Linear",
• "__torch__.chai.model.modules.af3_diffusion_transformer.___torch_mangle_1689.DiffusionTransformerBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1708.Linear",
• "__torch__.chai.model.modules.af3_diffusion_transformer.___torch_mangle_1721.DiffusionTransformerBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1704.Linear",
• "__torch__.torch.nn.modules.activation.___torch_mangle_1732.Sigmoid",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1593.Linear",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1504.TransitionLayer_v1",
• "__torch__.torch.nn.modules.container.___torch_mangle_1749.Sequential",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1839.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1852.Linear",
• "__torch__.torch.nn.modules.container.___torch_mangle_1873.ModuleList",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1596.Linear",
• "__torch__.chai.model.modules.af3_diffusion_transformer.___torch_mangle_1571.LocalDiffusionTransformer",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1549.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1794.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1670.LayerNorm",
• "__torch__.chai.model.blocks3.cond_transition.___torch_mangle_1663.ConditionedTransitionBlock_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1710.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_1875.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1680.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1773.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1861.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1712.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1805.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1534.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1724.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1532.Linear",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1665.AdaLayerNorm_v2",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1675.AdaLayerNorm_v2",
• "__torch__.chai.model.modules.af3_diffusion_transformer.DenseDiffusionTransformer",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1697.AdaLayerNorm_v2",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1810.Linear",
• "__torch__.torch.nn.modules.activation.___torch_mangle_1780.Sigmoid",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1806.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_1858.EinMix",
• "__torch__.chai.model.modules.af3_atom_attention_encoder_blocked.___torch_mangle_1529.AtomPairUpdateBlockedBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1520.Linear",
• "__torch__.torch.nn.modules.activation.___torch_mangle_1668.Sigmoid",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1693.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1646.Linear",
• "__torch__.torch.nn.modules.container.___torch_mangle_1701.Sequential",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1627.AdaLayerNorm_v2",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1783.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1648.Linear",
• "__torch__.chai.model.blocks3.cond_transition.___torch_mangle_1775.ConditionedTransitionBlock_v1",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1513.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1808.Linear",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_1872.LocalAttentionPairBiasBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1880.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1830.LayerNorm",
• "__torch__.torch.nn.modules.container.___torch_mangle_1492.Sequential",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1642.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1530.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1650.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1536.Linear",
• "__torch__.chai.model.modules.af3_diffusion_transformer.DiffusionTransformerBlock",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1750.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1690.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1586.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1766.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1645.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1702.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1821.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1831.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1594.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1600.Linear",
• "__torch__.torch.nn.modules.container.___torch_mangle_1489.Sequential",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1882.Linear",
• "__torch__.chai.model.blocks3.cond_transition.___torch_mangle_1853.ConditionedTransitionBlock_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1816.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1598.Linear",
• "__torch__.chai.model.modules.af3_diffusion_transformer.___torch_mangle_1657.DiffusionTransformerBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1667.Linear",
• "__torch__.torch.nn.modules.activation.___torch_mangle_1828.Sigmoid",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1832.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1610.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_1864.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1589.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1539.Linear",
• "__torch__.torch.nn.modules.activation.___torch_mangle_1519.ReLU",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1681.AdaLayerNorm_v2",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1752.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1736.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1628.Linear",
• "__torch__.torch.nn.modules.container.___torch_mangle_1854.ModuleList",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1761.AdaLayerNorm_v2",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1629.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1826.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1814.LayerNorm",
• "__torch__.chai.model.blocks3.cond_transition.___torch_mangle_1759.ConditionedTransitionBlock_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1774.Linear",
• "__torch__.chai.model.modules.af3_atom_attention_encoder_blocked.___torch_mangle_1579.AtomAttentionBlockedEncoder",
• "__torch__.torch.nn.modules.container.___torch_mangle_1717.Sequential",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1851.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1725.Linear",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1862.AdaLayerNorm_v2",
• "__torch__.einops.layers.torch.___torch_mangle_1857.Rearrange",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1771.AdaLayerNorm_v2",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1499.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1719.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1608.Linear",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_1860.LocalAttentionPairBiasBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1678.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1786.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1660.Linear",
• "__torch__.chai.model.blocks3.cond_transition.___torch_mangle_1841.ConditionedTransitionBlock_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1722.Linear",
• "__torch__.chai.model.modules.af3_diffusion_transformer.___torch_mangle_1877.LocalDiffusionTransformer",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1692.Linear",
• "__torch__.chai.model.blocks3.cond_transition.___torch_mangle_1791.ConditionedTransitionBlock_v1",
• "__torch__.torch.nn.modules.container.___torch_mangle_1797.Sequential",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1871.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1487.LayerNorm",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_1560.LocalAttentionPairBiasBlock",
• "__torch__.chai.model.blocks3.cond_transition.___torch_mangle_1679.ConditionedTransitionBlock_v1",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1809.AdaLayerNorm_v2",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1619.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1742.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1842.Linear",
• "__torch__.torch.nn.modules.container.___torch_mangle_1637.Sequential",
• "__torch__.torch.nn.modules.activation.___torch_mangle_1700.Sigmoid",
• "__torch__.chai.model.blocks3.cond_transition.___torch_mangle_1541.ConditionedTransitionBlock_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1838.Linear",
• "__torch__.chai.model.modules.af3_diffusion_transformer.___torch_mangle_1801.DiffusionTransformerBlock",
• "__torch__.chai.model.modules.af3_diffusion_transformer.___torch_mangle_1785.DiffusionTransformerBlock",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1501.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1502.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1767.Linear",
• "__torch__.torch.nn.modules.activation.___torch_mangle_1652.Sigmoid",
• "__torch__.torch.nn.modules.activation.Sigmoid",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1559.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1883.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1588.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1638.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1720.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1779.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1738.Linear",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1745.AdaLayerNorm_v2",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1782.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1784.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1493.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1757.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1662.Linear",
• "__torch__.torch.nn.modules.container.___torch_mangle_1881.Sequential",
• "__torch__.chai.model.blocks3.cond_transition.___torch_mangle_1535.ConditionedTransitionBlock_v1",
• "__torch__.chai.model.blocks3.cond_transition.___torch_mangle_1743.ConditionedTransitionBlock_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1763.Linear",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1837.AdaLayerNorm_v2",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1651.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1740.Linear",
• "__torch__.chai.model.modules.af3_diffusion_transformer.___torch_mangle_1753.DiffusionTransformerBlock",
• "__torch__.chai.model.blocks3.cond_transition.___torch_mangle_1807.ConditionedTransitionBlock_v1",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1691.AdaLayerNorm_v2",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1822.Linear",
• "__torch__.torch.nn.modules.activation.___torch_mangle_1684.Sigmoid",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1798.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1576.LayerNorm",
• "__torch__.torch.nn.modules.container.___torch_mangle_1781.Sequential",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1568.LayerNorm",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1601.AdaLayerNorm_v2",
• "__torch__.chai.model.modules.af3_atom_transformer.___torch_mangle_1878.AtomTransformer",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1546.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1603.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1699.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1744.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1709.Linear",
• "__torch__.torch.nn.modules.activation.___torch_mangle_1716.Sigmoid",
• "__torch__.chai.model.modules.af3_diffusion_transformer.___torch_mangle_1705.DiffusionTransformerBlock",
• "__torch__.torch.nn.modules.container.___torch_mangle_1570.Sequential",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1698.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1734.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1503.Linear",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1755.AdaLayerNorm_v2",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1543.AdaLayerNorm_v2",
• "__torch__.chai.model.blocks3.cond_transition.___torch_mangle_1647.ConditionedTransitionBlock_v1",
• "__torch__.einops.layers.torch.___torch_mangle_1563.Rearrange",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1498.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1855.Linear",
• "__torch__.chai.model.blocks3.cond_transition.___torch_mangle_1615.ConditionedTransitionBlock_v1",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1654.LayerNorm",
• "__torch__.chai.model.blocks3.cond_transition.___torch_mangle_1823.ConditionedTransitionBlock_v1",
• "__torch__.einops.layers.torch.___torch_mangle_1564.EinMix",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1707.AdaLayerNorm_v2",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_1554.LocalAttentionPairBiasBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1523.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1577.Linear",
• "__torch__.chai.model.blocks3.cond_transition.___torch_mangle_1695.ConditionedTransitionBlock_v1",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1509.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1820.Linear",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1723.AdaLayerNorm_v2",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1580.Linear",
• "__torch__.torch.nn.modules.activation.___torch_mangle_1604.Sigmoid",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1815.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1565.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1756.Linear",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1868.AdaLayerNorm_v2",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1611.AdaLayerNorm_v2",
• "__torch__.torch.nn.modules.activation.___torch_mangle_1636.Sigmoid",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1518.Linear",
• "__torch__.torch.nn.modules.activation.___torch_mangle_1812.Sigmoid",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1788.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1525.Linear",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1803.AdaLayerNorm_v2",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1824.Linear",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_1866.LocalAttentionPairBiasBlock",
• "__torch__.torch.nn.modules.activation.___torch_mangle_1574.ReLU",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1540.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1696.Linear",
• "__torch__.chai.model.modules.af3_fourier_emb.FourierEmbed",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1661.Linear",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1595.AdaLayerNorm_v2",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1819.AdaLayerNorm_v2",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1741.Linear",
• "__torch__.torch.nn.modules.container.___torch_mangle_1765.Sequential",
• "__torch__.chai.model.modules.af3_atom_transformer.___torch_mangle_1572.AtomTransformer",
• "__torch__.chai.model.blocks3.cond_transition.___torch_mangle_1847.ConditionedTransitionBlock_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1544.Linear",
• "__torch__.torch.nn.modules.container.___torch_mangle_1621.Sequential",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1630.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1506.Linear",
• "__torch__.torch.nn.modules.container.___torch_mangle_1578.Sequential",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1632.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1533.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1682.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1840.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1553.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1538.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1844.Linear",
• "__torch__.torch.nn.modules.container.___torch_mangle_1653.Sequential",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1776.Linear",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1649.AdaLayerNorm_v2",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1825.AdaLayerNorm_v2",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1703.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1614.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1747.Linear",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_1566.LocalAttentionPairBiasBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1583.Linear",
• "__torch__.torch.nn.modules.activation.___torch_mangle_1796.Sigmoid",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1495.Linear",
• "__torch__.chai.model.blocks3.cond_transition.___torch_mangle_1711.ConditionedTransitionBlock_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1602.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1505.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1730.Linear",
• "__torch__.chai.model.blocks3.cond_transition.___torch_mangle_1547.ConditionedTransitionBlock_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1846.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1879.LayerNorm",
• "__torch__.torch.nn.modules.container.___torch_mangle_1669.Sequential",
• "__torch__.torch.nn.modules.container.___torch_mangle_1829.Sequential",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1562.AdaLayerNorm_v2",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1850.Linear",
• "__torch__.torch.nn.modules.container.___torch_mangle_1876.Sequential",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1787.AdaLayerNorm_v2",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1768.Linear",
• "__torch__.torch.nn.modules.container.___torch_mangle_1813.Sequential",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1792.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1802.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1658.Linear",
• "torch._utils._rebuild_tensor_v2",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1731.Linear",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1550.AdaLayerNorm_v2",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1790.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1491.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1512.LayerNorm",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1591.LayerNorm",
• "__torch__.chai.model.blocks3.cond_transition.___torch_mangle_1727.ConditionedTransitionBlock_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1848.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_1558.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1612.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_1869.Rearrange",
• "__torch__.chai.model.modules.af3_diffusion_transformer.___torch_mangle_1625.DiffusionTransformerBlock",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1500.TransitionLayer_v1",
• "__torch__.torch.nn.modules.activation.___torch_mangle_1748.Sigmoid",
• "torch.FloatStorage",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1597.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1751.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1827.Linear",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1739.AdaLayerNorm_v2",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1497.LayerNorm",
• "__torch__.torch.nn.modules.container.___torch_mangle_1517.Sequential",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1496.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1664.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1804.Linear",
• "__torch__.torch.nn.modules.container.___torch_mangle_1590.Sequential",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1818.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1655.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_1551.Rearrange",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1688.Linear",
• "__torch__.chai.model.modules.af3_atom_attention_decoder.AtomAttentionDecoder",
• "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1508.TransitionLayer_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1726.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1640.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1706.Linear",
• "__torch__.torch.nn.modules.activation.___torch_mangle_1620.Sigmoid",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1676.Linear",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1659.AdaLayerNorm_v2",
• "__torch__.torch.nn.modules.activation.___torch_mangle_1764.Sigmoid",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1836.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1795.Linear",
• "__torch__.chai.model.blocks3.cond_transition.___torch_mangle_1631.ConditionedTransitionBlock_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1758.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1515.LayerNorm",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1633.AdaLayerNorm_v2",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1671.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1718.LayerNorm",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1587.AdaLayerNorm_v2",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1531.AdaLayerNorm_v2",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1799.Linear",
• "__torch__.chai.model.modules.af3_diffusion_transformer.___torch_mangle_1641.DiffusionTransformerBlock",
• "collections.OrderedDict",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1617.AdaLayerNorm_v2",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1545.Linear",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1843.AdaLayerNorm_v2",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1573.Linear",
• "__torch__.torch.nn.modules.container.___torch_mangle_1528.Sequential",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1849.AdaLayerNorm_v2",
• "__torch__.chai.model.modules.af3_diffusion_transformer.___torch_mangle_1769.DiffusionTransformerBlock",
• "__torch__.chai.model.modules.af3_diffusion_transformer.___torch_mangle_1817.DiffusionTransformerBlock",
• "__torch__.torch.nn.modules.container.___torch_mangle_1685.Sequential",
• "__torch__.chai.model.modules.af3_diffusion_transformer.___torch_mangle_1673.DiffusionTransformerBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1674.Linear",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1606.LayerNorm",
• "__torch__.torch.nn.modules.container.___torch_mangle_1511.Sequential",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1618.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1527.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_1557.Rearrange",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_1884.LayerNorm",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1746.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1516.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1510.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1672.Linear",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1856.AdaLayerNorm_v2",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1729.AdaLayerNorm_v2",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1537.AdaLayerNorm_v2",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1800.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1514.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_1863.Rearrange",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1607.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1555.Linear",
• "__torch__.torch.nn.modules.activation.___torch_mangle_1526.ReLU",
• "__torch__.torch.nn.modules.container.___torch_mangle_1733.Sequential",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1542.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1683.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1772.Linear",
• "__torch__.chai.model.modules.af3_diffusion_transformer.___torch_mangle_1737.DiffusionTransformerBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1762.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_1870.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1865.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1507.Linear",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1556.AdaLayerNorm_v2",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1714.Linear",
• "__torch__.chai.model.modules.af3_diffusion_module.DiffusionModule",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1634.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1687.Linear",
• "__torch__.chai.model.blocks3.cond_transition.___torch_mangle_1585.ConditionedTransitionBlock_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1789.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1867.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1666.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1582.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1754.Linear",
• "__torch__.torch.nn.modules.container.___torch_mangle_1567.ModuleList",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1715.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1613.Linear",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1793.AdaLayerNorm_v2",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1644.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1592.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1626.Linear",
• "__torch__.chai.model.modules.af3_diffusion_conditioning.DiffusionConditioning",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1713.AdaLayerNorm_v2",
• "__torch__.torch.nn.modules.linear.___torch_mangle_1778.Linear",
• "torch.LongStorage",
• "torch._utils._rebuild_tensor_v2",
• "torch.DoubleStorage",
• "collections.OrderedDict",
• "torch.FloatStorage",
• "collections.OrderedDict",
• "torch.LongStorage",
• "torch.BoolStorage",
• "torch.jit._pickle.restore_type_tag",
• "torch._utils._rebuild_tensor_v2"

How to fix it?

• "torch.LongStorage",
• "__torch__.torch.nn.modules.container.___torch_mangle_21.ModuleDict",
• "__torch__.torch.nn.modules.container.ModuleDict",
• "__torch__.chai.model.embedding.identity.___torch_mangle_10.Identity",
• "__torch__.chai.model.embedding.identity.___torch_mangle_1.Identity",
• "__torch__.chai.model.embedding.one_hot.___torch_mangle_7.OneHot",
• "__torch__.torch.nn.modules.sparse.Embedding",
• "__torch__.chai.model.embedding.one_hot.___torch_mangle_13.OneHot",
• "__torch__.chai.model.embedding.one_hot.OneHot",
• "__torch__.chai.model.embedding.identity.___torch_mangle_9.Identity",
• "__torch__.torch.nn.modules.container.___torch_mangle_43.ModuleDict",
• "__torch__.torch.nn.modules.linear.___torch_mangle_37.Linear",
• "__torch__.chai.model.embedding.one_hot.___torch_mangle_16.OneHot",
• "__torch__.chai.model.embedding.one_hot.___torch_mangle_15.OneHot",
• "__torch__.chai.model.embedding.identity.___torch_mangle_4.Identity",
• "__torch__.chai.model.embedding.rbf.RBF",
• "__torch__.chai.model.embedding.identity.___torch_mangle_25.Identity",
• "__torch__.torch.nn.modules.container.___torch_mangle_34.Sequential",
• "__torch__.torch.nn.modules.container.___torch_mangle_31.ModuleDict",
• "__torch__.chai.model.embedding.identity.___torch_mangle_22.Identity",
• "torch.FloatStorage",
• "__torch__.torch.nn.modules.linear.___torch_mangle_35.Linear",
• "__torch__.chai.model.embedding.one_hot.___torch_mangle_0.OneHot",
• "__torch__.chai.model.embedding.feature_embedding.FeatureEmbedding",
• "__torch__.chai.model.embedding.one_hot.___torch_mangle_23.OneHot",
• "torch._utils._rebuild_tensor_v2",
• "__torch__.chai.model.embedding.identity.Identity",
• "__torch__.chai.model.embedding.identity.___torch_mangle_8.Identity",
• "__torch__.torch.nn.modules.container.___torch_mangle_5.ModuleDict",
• "__torch__.chai.model.embedding.one_hot.___torch_mangle_11.OneHot",
• "__torch__.chai.model.embedding.identity.___torch_mangle_24.Identity",
• "__torch__.torch.nn.modules.linear.Linear",
• "__torch__.torch.nn.modules.container.___torch_mangle_42.Sequential",
• "__torch__.chai.model.embedding.esm.ESMEmbedding",
• "__torch__.torch.nn.modules.container.___torch_mangle_38.Sequential",
• "__torch__.torch.nn.modules.linear.___torch_mangle_39.Linear",
• "__torch__.chai.model.embedding.one_hot.___torch_mangle_17.OneHot",
• "__torch__.chai.model.embedding.one_hot.___torch_mangle_18.OneHot",
• "__torch__.torch.nn.modules.container.___torch_mangle_14.ModuleDict",
• "__torch__.torch.nn.modules.container.___torch_mangle_40.Sequential",
• "__torch__.torch.nn.modules.container.___torch_mangle_32.ModuleDict",
• "__torch__.torch.nn.modules.container.Sequential",
• "collections.OrderedDict",
• "__torch__.torch.nn.modules.container.___torch_mangle_36.Sequential",
• "__torch__.chai.model.embedding.one_hot.___torch_mangle_26.OneHot",
• "__torch__.torch.nn.modules.linear.___torch_mangle_41.Linear",
• "__torch__.torch.nn.modules.container.___torch_mangle_27.ModuleDict",
• "__torch__.chai.model.embedding.identity.___torch_mangle_29.Identity",
• "__torch__.chai.model.embedding.one_hot.___torch_mangle_28.OneHot",
• "__torch__.chai.model.embedding.rbf.___torch_mangle_20.RBF",
• "__torch__.chai.model.embedding.one_hot.___torch_mangle_3.OneHot",
• "__torch__.chai.model.embedding.one_hot.___torch_mangle_19.OneHot",
• "__torch__.chai.model.embedding.one_hot.___torch_mangle_12.OneHot",
• "__torch__.chai.model.embedding.identity.___torch_mangle_30.Identity",
• "__torch__.torch.nn.modules.linear.___torch_mangle_33.Linear",
• "__torch__.chai.model.embedding.identity.___torch_mangle_6.Identity",
• "__torch__.chai.model.embedding.identity.___torch_mangle_2.Identity",
• "__torch__.chai.model.embedding.outer_sum.OuterSumProj",
• "torch.LongStorage",
• "torch.FloatStorage",
• "torch._utils._rebuild_tensor_v2",
• "torch.DoubleStorage",
• "collections.OrderedDict",
• "torch.jit._pickle.restore_type_tag",
• "torch.BFloat16Storage",
• "collections.OrderedDict",
• "torch._utils._rebuild_tensor_v2"

How to fix it?

• "__torch__.chai.model.modules.af3_attention_pair_bias_block.LocalAttentionPairBiasBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_48.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_70.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_82.EinMix",
• "__torch__.torch.nn.modules.container.___torch_mangle_53.Sequential",
• "__torch__.torch.nn.modules.activation.___torch_mangle_89.ReLU",
• "__torch__.torch.nn.modules.linear.___torch_mangle_94.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_86.EinMix",
• "__torch__.chai.model.modules.af3_atom_transformer.AtomTransformer",
• "__torch__.torch.nn.modules.linear.___torch_mangle_77.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_64.Linear",
• "__torch__.torch.nn.modules.container.___torch_mangle_90.Sequential",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_71.AdaLayerNorm_v2",
• "__torch__.torch.nn.modules.container.___torch_mangle_85.ModuleList",
• "__torch__.torch.nn.modules.linear.___torch_mangle_68.Linear",
• "torch.FloatStorage",
• "__torch__.torch.nn.modules.linear.___torch_mangle_91.Linear",
• "__torch__.chai.model.blocks3.ada_layernorm.AdaLayerNorm_v2",
• "__torch__.chai.model.modules.af3_diffusion_transformer.LocalDiffusionTransformer",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_84.LocalAttentionPairBiasBlock",
• "__torch__.torch.nn.modules.normalization.LayerNorm",
• "__torch__.torch.nn.modules.activation.___torch_mangle_47.ReLU",
• "torch._utils._rebuild_tensor_v2",
• "__torch__.torch.nn.modules.container.ModuleList",
• "__torch__.chai.model.blocks3.cond_transition.___torch_mangle_63.ConditionedTransitionBlock_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_60.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_66.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_56.Linear",
• "__torch__.einops.layers.torch.Rearrange",
• "__torch__.torch.nn.modules.linear.___torch_mangle_83.Linear",
• "__torch__.chai.model.af3.token_input_emb.TokenInputEmbedding",
• "__torch__.chai.model.blocks3.cond_transition.___torch_mangle_69.ConditionedTransitionBlock_v1",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_74.AdaLayerNorm_v2",
• "__torch__.chai.model.modules.af3_input_embedder.AF3InputEmbedder",
• "__torch__.torch.nn.modules.linear.___torch_mangle_57.Linear",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_80.AdaLayerNorm_v2",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_59.AdaLayerNorm_v2",
• "__torch__.torch.nn.modules.linear.___torch_mangle_50.Linear",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_78.LocalAttentionPairBiasBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_55.Linear",
• "__torch__.chai.model.modules.af3_atom_attention_encoder_blocked.AtomPairUpdateBlockedBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_62.Linear",
• "__torch__.einops.layers.torch.EinMix",
• "__torch__.torch.nn.modules.activation.ReLU",
• "collections.OrderedDict",
• "__torch__.torch.nn.modules.linear.___torch_mangle_79.Linear",
• "__torch__.chai.model.modules.af3_atom_attention_encoder_blocked.AtomAttentionBlockedEncoder",
• "__torch__.torch.nn.modules.linear.___torch_mangle_92.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_61.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_72.Linear",
• "__torch__.chai.model.blocks3.cond_transition.ConditionedTransitionBlock_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_67.Linear",
• "__torch__.torch.nn.modules.container.___torch_mangle_87.Sequential",
• "__torch__.torch.nn.modules.linear.___torch_mangle_54.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_76.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_58.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_88.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_52.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_73.Linear",
• "__torch__.torch.nn.modules.activation.___torch_mangle_51.ReLU",
• "__torch__.einops.layers.torch.___torch_mangle_75.Rearrange",
• "__torch__.torch.nn.modules.linear.___torch_mangle_44.Linear",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_65.AdaLayerNorm_v2",
• "__torch__.torch.nn.modules.linear.___torch_mangle_93.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_81.Rearrange",
• "__torch__.torch.nn.modules.container.___torch_mangle_46.Sequential",
• "__torch__.torch.nn.modules.container.___torch_mangle_49.Sequential",
• "__torch__.torch.nn.modules.linear.___torch_mangle_45.Linear",
• "collections.OrderedDict",
• "torch._utils._rebuild_tensor_v2",
• "torch.LongStorage",
• "torch.LongStorage",
• "torch.BFloat16Storage",
• "torch.jit._pickle.restore_type_tag",
• "torch._utils._rebuild_tensor_v2",
• "collections.OrderedDict",
• "torch.BoolStorage"

How to fix it?

• "__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?