Deep PIP Kernels
================

.. automodule:: molpipx.pipgp_flax
   :members:
   :exclude-members: base_kernel, network, dummy_x, key, nn_params, compute_engine
   :undoc-members:
   :show-inheritance:

   .. rubric:: Classes

   .. autosummary::
      :nosignatures:

      PIPLayerKernel