Ximinf Documentation
====================

Welcome to **ximinf** — Simulation-Based Inference of Cosmological Parameters in JAX using Type Ia Supernovae.

This documentation will guide you through installation, usage, and the API reference.

Contents
--------

.. toctree::
   :maxdepth: 2
   :caption: "Documentation"

Quickstart
----------

.. code-block:: python

    import ximinf


API Reference
-------------

.. automodule:: ximinf.generate_sim
    :members:
    :undoc-members:
    :show-inheritance:

.. automodule:: ximinf.nn_inference
    :members:
    :undoc-members:

.. automodule:: ximinf.nn_train
    :members:
    :undoc-members:

.. automodule:: ximinf.nn_test
    :members:
    :undoc-members:

.. automodule:: ximinf.selection_effects
    :members:
    :undoc-members: