Metadata-Version: 2.4
Name: jax-moseq
Version: 0.3.5
Home-page: https://github.com/dattalab/jax-moseq/
Author: Caleb Weinreb
Author-email: calebsw@gmail.com
License: LicenseRef-Harvard-Academic-Non-Commercial
Classifier: Programming Language :: Python :: 3
Classifier: Operating System :: OS Independent
Requires-Python: >=3.10
Description-Content-Type: text/markdown
License-File: LICENSE.md
Requires-Dist: numba>=0.56.4
Requires-Dist: dynamax>=0.1.2
Requires-Dist: tqdm
Requires-Dist: optree
Provides-Extra: cuda
Requires-Dist: jax[cuda12]; extra == "cuda"
Dynamic: description
Dynamic: description-content-type
Dynamic: license-file

# 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 srdatta@hms.harvard.edu, who will put you in touch with the appropriate people in the Harvard Technology Transfer office.

