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

This section contains the complete API reference for metalog-jax.

.. contents:: Table of Contents
   :local:
   :depth: 2

metalog_jax
-----------

Main Module
~~~~~~~~~~~

.. automodule:: metalog_jax
   :members:
   :undoc-members:
   :show-inheritance:

Metalog Distribution
~~~~~~~~~~~~~~~~~~~~

.. automodule:: metalog_jax.metalog
   :members:
   :undoc-members:
   :show-inheritance:

Grid Search
~~~~~~~~~~~

.. automodule:: metalog_jax.grid_search
   :members:
   :undoc-members:
   :show-inheritance:

Regression
----------

Base Classes
~~~~~~~~~~~~

.. automodule:: metalog_jax.regression.base
   :members:
   :undoc-members:
   :show-inheritance:

Ordinary Least Squares
~~~~~~~~~~~~~~~~~~~~~~

.. automodule:: metalog_jax.regression.ols
   :members:
   :undoc-members:
   :show-inheritance:

LASSO
~~~~~

.. automodule:: metalog_jax.regression.lasso
   :members:
   :undoc-members:
   :show-inheritance: