xnn.gnn.models.aimnet2_foundation.from_aimnet_artifact#

xnn.gnn.models.aimnet2_foundation.from_aimnet_artifact(artifact, dtype=None)[source]#

Convert a reference AIMNet2 artifact to an xnn AIMNet2.

Parameters:

artifact (Mapping) – As returned by load_artifact().
dtype (torch.dtype or str, optional) – Final dtype; None keeps the artifact’s float32.

Returns:

The bare network (no dispersion term); its D3 record is available through convert_artifact().

Return type:

AIMNet2