deepmd.jax
==========

.. py:module:: deepmd.jax

.. autoapi-nested-parse::

   
   JAX backend.
















   ..
       !! processed by numpydoc !!


Submodules
----------

.. toctree::
   :maxdepth: 1

   /autoapi/deepmd/jax/atomic_model/index
   /autoapi/deepmd/jax/common/index
   /autoapi/deepmd/jax/descriptor/index
   /autoapi/deepmd/jax/env/index
   /autoapi/deepmd/jax/fitting/index
   /autoapi/deepmd/jax/infer/index
   /autoapi/deepmd/jax/jax2tf/index
   /autoapi/deepmd/jax/model/index
   /autoapi/deepmd/jax/utils/index


Functions
---------

.. autoapisummary::

   deepmd.jax.load_entry_point


Package Contents
----------------

.. py:function:: load_entry_point(group: str) -> list

   
   Load entry points from a group.


   :Parameters:

       **group** : :class:`python:str`
           The group name.



   :Returns:

       :class:`python:list`
           A list of loaded entry points.











   ..
       !! processed by numpydoc !!