deepmd.tf2

TensorFlow eager backend based on dpmodel Array API code.

Submodules

• deepmd.tf2.atomic_model
• deepmd.tf2.common
• deepmd.tf2.descriptor
• deepmd.tf2.entrypoints
• deepmd.tf2.env
• deepmd.tf2.fitting
• deepmd.tf2.infer
• deepmd.tf2.model
• deepmd.tf2.modifier
• deepmd.tf2.train
• deepmd.tf2.transform_output
• deepmd.tf2.utils