API Reference#
This page contains auto-generated API reference documentation [1].
- core
- core.common
- core.common.relaxation
- core.common.data_parallel
- core.common.distutils
- core.common.flags
- core.common.gp_utils
- core.common.hpo_utils
- core.common.logger
- core.common.profiler_utils
- core.common.registry
- core.common.slurm
- core.common.test_utils
- core.common.transforms
- core.common.tutorial_utils
- core.common.typing
- core.common.utils
- core.datasets
- core.models
- core.models.equiformer_v2
- core.models.equiformer_v2.prediction_heads
- core.models.equiformer_v2.trainers
- core.models.equiformer_v2.activation
- core.models.equiformer_v2.drop
- core.models.equiformer_v2.edge_rot_mat
- core.models.equiformer_v2.equiformer_v2
- core.models.equiformer_v2.equiformer_v2_deprecated
- core.models.equiformer_v2.eqv2_to_eqv2_hydra
- core.models.equiformer_v2.gaussian_rbf
- core.models.equiformer_v2.input_block
- core.models.equiformer_v2.layer_norm
- core.models.equiformer_v2.module_list
- core.models.equiformer_v2.radial_function
- core.models.equiformer_v2.so2_ops
- core.models.equiformer_v2.so3
- core.models.equiformer_v2.transformer_block
- core.models.equiformer_v2.wigner
- core.models.escn
- core.models.gemnet
- core.models.gemnet.layers
- core.models.gemnet.layers.atom_update_block
- core.models.gemnet.layers.base_layers
- core.models.gemnet.layers.basis_utils
- core.models.gemnet.layers.efficient
- core.models.gemnet.layers.embedding_block
- core.models.gemnet.layers.interaction_block
- core.models.gemnet.layers.radial_basis
- core.models.gemnet.layers.spherical_basis
- core.models.gemnet.gemnet
- core.models.gemnet.initializers
- core.models.gemnet.utils
- core.models.gemnet.layers
- core.models.gemnet_gp
- core.models.gemnet_gp.layers
- core.models.gemnet_gp.layers.atom_update_block
- core.models.gemnet_gp.layers.base_layers
- core.models.gemnet_gp.layers.basis_utils
- core.models.gemnet_gp.layers.efficient
- core.models.gemnet_gp.layers.embedding_block
- core.models.gemnet_gp.layers.interaction_block
- core.models.gemnet_gp.layers.radial_basis
- core.models.gemnet_gp.layers.spherical_basis
- core.models.gemnet_gp.gemnet
- core.models.gemnet_gp.initializers
- core.models.gemnet_gp.utils
- core.models.gemnet_gp.layers
- core.models.gemnet_oc
- core.models.gemnet_oc.layers
- core.models.gemnet_oc.layers.atom_update_block
- core.models.gemnet_oc.layers.base_layers
- core.models.gemnet_oc.layers.basis_utils
- core.models.gemnet_oc.layers.efficient
- core.models.gemnet_oc.layers.embedding_block
- core.models.gemnet_oc.layers.force_scaler
- core.models.gemnet_oc.layers.interaction_block
- core.models.gemnet_oc.layers.radial_basis
- core.models.gemnet_oc.layers.spherical_basis
- core.models.gemnet_oc.gemnet_oc
- core.models.gemnet_oc.initializers
- core.models.gemnet_oc.interaction_indices
- core.models.gemnet_oc.utils
- core.models.gemnet_oc.layers
- core.models.painn
- core.models.scn
- core.models.utils
- core.models.base
- core.models.dimenet_plus_plus
- core.models.model_registry
- core.models.schnet
- core.models.equiformer_v2
- core.modules
- core.preprocessing
- core.scripts
- core.scripts.hpo
- core.scripts.download_data
- core.scripts.download_large_files
- core.scripts.eqv2_to_hydra_eqv2
- core.scripts.fit_normalizers
- core.scripts.fit_references
- core.scripts.gif_maker_parallelized
- core.scripts.make_challenge_submission_file
- core.scripts.make_lmdb_sizes
- core.scripts.make_submission_file
- core.scripts.preprocess_ef
- core.scripts.preprocess_relaxed
- core.scripts.uncompress
- core.tasks
- core.trainers
- core._cli
- core.common
- data
- ocpapi
- cattsunami