microsoft

/

simpoly

like 0

Follow

Microsoft 19.8k

• "__builtin__.set",
• "mlff.train.mliap.MLFFModelMLIAP",
• "torch.nn.modules.container.ModuleList",
• "cuequivariance.segmented_tensor_product.operand.Operand",
• "e3nn.o3._irreps.Irreps",
• "cuequivariance_torch.primitives.equivariant_tensor_product.SymmetricTPDispatcher",
• "numpy.core.multiarray._reconstruct",
• "torch.nn.modules.normalization.LayerNorm",
• "torch._utils._rebuild_tensor_v2",
• "cuequivariance.equivariant_tensor_product.EquivariantTensorProduct",
• "torch.device",
• "mlff.models.modules.atomic.AtomicEmbedding",
• "mlff.train.timers.CPUTimer",
• "cuequivariance_ops_torch.symmetric_tensor_contraction.SymmetricContractionBwdBwdInfo",
• "torch._utils._rebuild_parameter",
• "cuequivariance_torch.primitives.tensor_product.TensorProduct",
• "torch.IntStorage",
• "cuequivariance_ops_torch.tensor_product_uniform_1d.TensorProductUniform1d",
• "mlff.models.modules.ltp_cuequivariance.LinearWrapperTwist",
• "cuequivariance.irreps_array.irreps.MulIrrep",
• "mlff.models.vivace.explosion.VivaceBergamot",
• "e3nn.o3._irreps._MulIr",
• "torch.nn.modules.linear.Identity",
• "cuequivariance_ops_torch.symmetric_tensor_contraction.SymmetricContractionFwdInfo",
• "mlff.models.modules.ltp_cuequivariance.LTPCuEquivarianceTwist",
• "torch.fx.graph_module.reduce_graph_module",
• "mlff.models.modules.ltp_cuequivariance.MakeWeightedChannelsTwist",
• "torch.float32",
• "torch.nn.modules.activation.SiLU",
• "cuequivariance.irreps_array.irreps_layout.IrrepsLayout",
• "mlff.models.vivace.equivariant.MockInnerProduct",
• "numpy.ndarray",
• "cuequivariance_torch.primitives.transpose.TransposeIrrepsLayout",
• "cuequivariance_torch.primitives.symmetric_tensor_product.IWeightedSymmetricTensorProduct",
• "mlff.models.vivace.layers.InitHeader",
• "cuequivariance.irreps_array.irreps_and_layout.IrrepsAndLayout",
• "mlff.models.vivace.equivariant.VivaceTensorProduct",
• "collections.OrderedDict",
• "mlff.models.modules.radial_clamping.CosineClamp",
• "cuequivariance.segmented_tensor_product.subscripts.Subscripts",
• "cuequivariance_torch.operations.spherical_harmonics.SphericalHarmonics",
• "cuequivariance.segmented_tensor_product.segmented_tensor_product.SegmentedTensorProduct",
• "numpy.dtype",
• "cuequivariance.irreps_array.irreps.Irreps",
• "mlff.models.modules.ltp_cuequivariance.DotProductCuEquivarianceTwist",
• "cuequivariance_torch.primitives.equivariant_tensor_product.Transpose1Dispatcher",
• "mlff.models.vivace.equivariant.InvariantCrossAttention2EquivariantNode",
• "mlff.models.modules.radial_basis_set.ExpNormalSmearing",
• "cuequivariance_torch.primitives.tensor_product._TwoArgCaller",
• "e3nn.o3._irreps.Irrep",
• "mlff.models.vivace.layers.EnergyOutput",
• "mlff.models.modules.attention.QKVExp",
• "cuequivariance.segmented_tensor_product.path.Path",
• "cuequivariance.representation.irrep_so3.SO3",
• "cuequivariance_torch.primitives.equivariant_tensor_product.EquivariantTensorProduct",
• "cuequivariance_torch.primitives.symmetric_tensor_product.SymmetricTensorProduct",
• "torch.nn.modules.linear.Linear",
• "cuequivariance_torch.primitives.symmetric_tensor_product.CUDAKernel",
• "_codecs.encode",
• "torch.FloatStorage",
• "cuequivariance_torch.primitives.tensor_product._Wrapper",
• "cuequivariance_ops_torch.symmetric_tensor_contraction.SymmetricTensorContraction",
• "cuequivariance_torch.primitives.tensor_product.TensorProductUniform3x1d",
• "mlff.models.modules.no_bias_mlp.NoBiasMLP",
• "collections.deque",
• "mlff.models.modules.mlp.MLP",
• "mlff.models.modules.radial_clamping.StepClamp",
• "cuequivariance_ops_torch.symmetric_tensor_contraction.SymmetricContractionBwdInfo"

How to fix it?