sequince.jax#
JAX-specific helpers for Sequince’s functional SMC interface.
Attributes#
Functions#
|
Lower and compile one stable functional iteration signature. |
Module Contents#
- sequince.jax.logger#
- sequince.jax.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.
- Parameters:
initial_state (sequince.state.ParticleState)
random_state (Any)
observation (Any)
next_observation_index (int)
shared_context (collections.abc.Mapping[str, Any] | None)
metadata (collections.abc.Mapping[str, Any] | None)
- Return type:
sequince.smc.IterationCallable