deepmd.jax.utils.update_sel

deepmd.jax.utils.update_sel#

Attributes#

_MISSING

Classes#

UpdateSel

Update the sel field in the descriptor.

Functions#

_get_update_sel_descriptors(→ tuple[type[Any], Ellipsis])

use_jax_update_sel(→ collections.abc.Iterator[None])

Use JAX neighbor statistics in dpmodel descriptor update_sel methods.

Module Contents#

class deepmd.jax.utils.update_sel.UpdateSel[source]#

Bases: deepmd.utils.update_sel.BaseUpdateSel

Update the sel field in the descriptor.

property neighbor_stat: type[deepmd.jax.utils.neighbor_stat.NeighborStat][source]#
deepmd.jax.utils.update_sel._MISSING[source]#
deepmd.jax.utils.update_sel._get_update_sel_descriptors() → tuple[type[Any], Ellipsis][source]#
deepmd.jax.utils.update_sel.use_jax_update_sel() → collections.abc.Iterator[None][source]#

Use JAX neighbor statistics in dpmodel descriptor update_sel methods.