Anisotropic PIPs and Models
===========================

.. automodule:: molpipx.pip_anisotropic_flax
   :members:
   :exclude-members: f_mono, f_poly, f_mask, l, n_pairs, bias_init, vmap, jit, lax
   :undoc-members:
   :show-inheritance:

   .. rubric:: Functions

   .. autosummary::
      :nosignatures:

      get_mask
      get_f_mask
      lambda_random_init

   .. rubric:: Classes

   .. autosummary::
      :nosignatures:

      PIPAniso
      LayerPIPAniso
      EnergyPIPAniso