sajax
=====

.. py:module:: sajax

.. autoapi-nested-parse::

   SAJAX — Stellar Activity Grid for Exoplanets in JAX.

   Public API
   ----------
   build_model
       Pre-build all static model arrays once before MCMC sampling.

   evaluate_light_curve
       Pure JAX evaluation — accepts JAX tracers, compatible with
       jit, vmap, emcee_jax, and gradient-based samplers.

   compute_light_curve
       Convenience wrapper: build_model + evaluate_light_curve in one call.
       Use for one-off evaluations outside MCMC.

   build_stellar_grid
       Pre-compute the static stellar pixel grid.

   rotate_active_region
       Apply stellar rotation and inclination to a Cartesian active region
       position.

   _compute_planet_mask
       Compute the mask over stellar disc pixels: ``True`` where the pixel is occulted
       by the planet at this epoch.

   LdcMode
       Type alias for supported limb-darkening laws.

   ArOverlapMode
       Type alias for active-region overlap resolution rules.



Submodules
----------

.. toctree::
   :maxdepth: 1

   /autoapi/sajax/core/index
   /autoapi/sajax/geometry/index
   /autoapi/sajax/planet/index