Keypoint-SLDS
=============


Initialization
--------------

.. automodule:: jax_moseq.models.keypoint_slds.initialize
   :members:


Gibbs sampling
--------------

.. automodule:: jax_moseq.models.keypoint_slds.gibbs
   :members:


Log probability
---------------

.. automodule:: jax_moseq.models.keypoint_slds.log_prob
   :members:


Geometric transforms
--------------------

.. automodule:: jax_moseq.models.keypoint_slds.alignment
   :members: