JAX Alignment
=============

Sequential JAX-based alignment functions.  For **multi-device parallel**
alignment via ``jax.shard_map`` see :doc:`jax_parallel`.

.. automodule:: shepherd_score.alignment._jax
   :members:
   :undoc-members:
   :show-inheritance: