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

DiRe-JAX is a high-performance dimensionality reduction library built on JAX, designed for efficient processing of large-scale, high-dimensional datasets.

DiRe JAX Overview
-----------------

.. automodule:: dire_jax
   :members:
   :undoc-members:
   :show-inheritance:
   :noindex:


Core Components
---------------

DiRe Class
~~~~~~~~~~

.. automodule:: dire_jax.dire
   :members:
   :undoc-members:
   :show-inheritance:
   :noindex:

Utility Functions
~~~~~~~~~~~~~~~~~

.. automodule:: dire_jax.dire_utils
   :members:
   :undoc-members:
   :show-inheritance:
   :noindex:

Kernalized kNN index
~~~~~~~~~~~~~~~~~~~~

.. automodule:: dire_jax.hpindex
   :members:
   :undoc-members:
   :show-inheritance:
   :noindex:

Performance Metrics
~~~~~~~~~~~~~~~~~~~

.. automodule:: dire_jax.hpmetrics
   :members:
   :undoc-members:
   :show-inheritance:
   :noindex: