API Reference
=============

This section documents the API for :code:`cebmf_torch`.

Main Package
------------

.. automodule:: cebmf_torch
   :members:
   :undoc-members:
   :no-index:

Core Classes
------------

.. autoclass:: cebmf_torch.cEBMF
   :members:
   :undoc-members:
   :show-inheritance:

EBNM Solvers
------------

.. automodule:: cebmf_torch.ebnm
   :members:
   :undoc-members:

.. autofunction:: cebmf_torch.ash

.. autofunction:: cebmf_torch.ebnm_point_exp

.. autofunction:: cebmf_torch.ebnm_point_laplace

Covariate-Enhanced EBNM
------------------------

.. automodule:: cebmf_torch.cebnm
   :members:
   :undoc-members:

Utilities
---------

.. automodule:: cebmf_torch.utils
   :members:
   :undoc-members:

Mathematical Functions
~~~~~~~~~~~~~~~~~~~~~~

.. automodule:: cebmf_torch.utils.maths
   :members:
   :undoc-members:

Device Management
~~~~~~~~~~~~~~~~~

.. automodule:: cebmf_torch.utils.device
   :members:
   :undoc-members:

Prior Registry
--------------

.. automodule:: cebmf_torch.priors
   :members:
   :undoc-members: