sequince.jax
============

.. py:module:: sequince.jax

.. autoapi-nested-parse::

   JAX-specific helpers for Sequince's functional SMC interface.



Attributes
----------

.. autoapisummary::

   sequince.jax.logger


Functions
---------

.. autoapisummary::

   sequince.jax.compile_functional_iteration


Module Contents
---------------

.. py:data:: logger

.. py:function:: compile_functional_iteration(smc, initial_state, random_state, *, observation = None, next_observation_index = 0, shared_context = None, metadata = None)

   Lower and compile one stable functional iteration signature.

   Functional state and context initialization runs before lowering, but
   compilation does not execute or discard an SMC transition. The supplied
   context structures must match those used by the subsequent run.