deepmd_gnn
==========

.. py:module:: deepmd_gnn

.. autoapi-nested-parse::

   MACE plugin for DeePMD-kit.



Submodules
----------

.. toctree::
   :maxdepth: 1

   /autoapi/deepmd_gnn/__main__/index
   /autoapi/deepmd_gnn/_version/index
   /autoapi/deepmd_gnn/argcheck/index
   /autoapi/deepmd_gnn/cli/index
   /autoapi/deepmd_gnn/env/index
   /autoapi/deepmd_gnn/mace/index
   /autoapi/deepmd_gnn/mace_off/index
   /autoapi/deepmd_gnn/mace_off_cli/index
   /autoapi/deepmd_gnn/nequip/index
   /autoapi/deepmd_gnn/op/index


Attributes
----------

.. autoapisummary::

   deepmd_gnn.__version__


Functions
---------

.. autoapisummary::

   deepmd_gnn.mace_model_args


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

.. py:data:: __version__
   :type:  str

.. py:function:: mace_model_args() -> dargs.Argument

   Arguments for the MACE model.

   Returns
   -------
   Argument
       Arguments for the MACE model.