API Reference

Common Utilities

`mxtaltools.common.adaptive_batching`
`mxtaltools.common.ase_interface`
`mxtaltools.common.clustering`
`mxtaltools.common.config_processing`
`mxtaltools.common.ellipsoid_ops`
`mxtaltools.common.geometry_utils`
`mxtaltools.common.mol_classifier_utils`
`mxtaltools.common.sym_utils`
`mxtaltools.common.training_utils`
`mxtaltools.common.utils`

`mxtaltools.constants.asymmetric_units`
`mxtaltools.constants.atom_properties`
`mxtaltools.constants.classifier_constants`
`mxtaltools.constants.csd_stats`
`mxtaltools.constants.mol_classifier_constants`
`mxtaltools.constants.space_group_info`
`mxtaltools.constants.space_group_reference`

`mxtaltools.crystal_building.crystal_latent_transforms`
`mxtaltools.crystal_building.utils`

`mxtaltools.dataset_utils.dataset_manager`
`mxtaltools.dataset_utils.utils`

Core model utilities:

`mxtaltools.models.autoencoder_utils`
`mxtaltools.models.utils`

Graph functions:

`mxtaltools.models.functions.minimum_image_neighbors`
`mxtaltools.models.functions.radial_graph`

Graph neural networks:

`mxtaltools.models.graph_models.base_graph_model`
`mxtaltools.models.graph_models.graph_neural_network`
`mxtaltools.models.graph_models.molecule_graph_model`

Network modules:

`mxtaltools.models.modules.augmented_softmax_aggregator`
`mxtaltools.models.modules.basis_functions`
`mxtaltools.models.modules.components`
`mxtaltools.models.modules.graph_convolution`
`mxtaltools.models.modules.vector_LayerNorm`

Task-specific models:

`mxtaltools.models.task_models.autoencoder_models`
`mxtaltools.models.task_models.crystal_models`
`mxtaltools.models.task_models.embedding_regression_models`
`mxtaltools.models.task_models.mol_classifier`
`mxtaltools.models.task_models.regression_models`