Skip to content

dattalab/jax-moseq

Repository files navigation

JAX MoSeq

Code for fitting state-space models with Gibbs sampling.

Installation

pip install jax-moseq # CPU
pip install jax-moseq[cuda] # GPU (CUDA 12)

License

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 [email protected], who will put you in touch with the appropriate people in the Harvard Technology Transfer office.