deepmd.jax.atomic_model.dp_atomic_model

Contents

deepmd.jax.atomic_model.dp_atomic_model#

Classes#

DPAtomicModel

Functions#

make_jax_dp_atomic_model_from_dpmodel(...)

Make a JAX backend DP atomic model from a DPModel backend DP atomic model.

Module Contents#

deepmd.jax.atomic_model.dp_atomic_model.make_jax_dp_atomic_model_from_dpmodel(dpmodel_atomic_model: type[deepmd.dpmodel.atomic_model.dp_atomic_model.DPAtomicModel]) → type[deepmd.dpmodel.atomic_model.dp_atomic_model.DPAtomicModel][source]#

Make a JAX backend DP atomic model from a DPModel backend DP atomic model.

Parameters:

dpmodel_atomic_modeltype[DPAtomicModelDP]: The DPModel backend DP atomic model.

Returns:

type[DPAtomicModel]: The JAX backend DP atomic model.

class deepmd.jax.atomic_model.dp_atomic_model.DPAtomicModel[source]#: Bases: make_jax_dp_atomic_model_from_dpmodel(DPAtomicModelDP)