AR-HMM
======

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

.. automodule:: jax_moseq.models.arhmm.initialize
   :members:

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

.. automodule:: jax_moseq.models.arhmm.gibbs
   :members:


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

.. automodule:: jax_moseq.models.arhmm.log_prob
   :members: