Utilities
=========

Distributions
-------------

.. automodule:: jax_moseq.utils.distributions
   :members:

Transitions
-----------

.. automodule:: jax_moseq.utils.transitions
   :members:

Kalman smoothing
----------------

.. automodule:: jax_moseq.utils.kalman
   :members:

Autoregression
--------------

.. automodule:: jax_moseq.utils.autoregression
   :members:

Odds and ends
-------------

.. automodule:: jax_moseq.utils.utils
   :members: