deepmd.jax.atomic_model.energy_atomic_model

Contents

deepmd.jax.atomic_model.energy_atomic_model#

Classes#

DPAtomicModelEnergy

Module Contents#

class deepmd.jax.atomic_model.energy_atomic_model.DPAtomicModelEnergy[source]#: Bases: make_jax_dp_atomic_model_from_dpmodel(DPAtomicModelEnergyDP)