eztaox.kernels.eqx_utils
========================

.. py:module:: eztaox.kernels.eqx_utils

.. autoapi-nested-parse::

   Utility methods for Equinox modules.



Functions
---------

.. autoapisummary::

   eztaox.kernels.eqx_utils.find_param_by_name


Module Contents
---------------

.. py:function:: find_param_by_name(module: equinox.Module, name: str) -> list | None

   Find a leaf parameter in an Equinox module by name.

   :param module: The Equinox module to search in.
   :type module: eqx.Module
   :param name: The name of the parameter to find.
   :type name: str

   :returns: The parameter if found, None otherwise.
   :rtype: list | None