deepmd.jax

JAX backend.

Submodules

• deepmd.jax.atomic_model
• deepmd.jax.common
• deepmd.jax.descriptor
• deepmd.jax.env
• deepmd.jax.fitting
• deepmd.jax.infer
• deepmd.jax.jax2tf
• deepmd.jax.model
• deepmd.jax.utils

Functions

load_entry_point(→ list)

Load entry points from a group.

Package Contents

deepmd.jax.load_entry_point(group: str) → list

Load entry points from a group.

Parameters:

groupstr

The group name.

Returns:

list

A list of loaded entry points.