Code for fitting state-space models with Gibbs sampling.
Install jax as described in Keypoint MoSeq docs, then install jax-moseq:
pip install jax-moseq
MoSeq is freely available for academic use under a license provided by Harvard University. Please refer to the license file for details. If you are interested in using MoSeq for commercial purposes please contact Bob Datta directly at srdatta@hms.harvard.edu, who will put you in touch with the appropriate people in the Harvard Technology Transfer office.