Jax-moseq

What's Changed
* Dynamax + bugfixes by calebweinreb in https://github.com/dattalab/jax-moseq/pull/27
* Lgssm dynamax parallel inference by kaijfox in https://github.com/dattalab/jax-moseq/pull/23

New Contributors
* kaijfox made their first contribution in https://github.com/dattalab/jax-moseq/pull/23

**Full Changelog**: https://github.com/dattalab/jax-moseq/compare/0.0.3...0.0.4

What's Changed
* debugging utils, determinism, side effect removal by calebweinreb in https://github.com/dattalab/jax-moseq/pull/20
* remove more side effects by calebweinreb in https://github.com/dattalab/jax-moseq/pull/21

**Full Changelog**: https://github.com/dattalab/jax-moseq/compare/0.0.2...0.0.3

What's Changed
* ensure discrete state dtype is int64/int32 depending on global config by calebweinreb in https://github.com/dattalab/jax-moseq/pull/19
* Added util that looks for nans in pytree. This is called by keypoint-moseq during model fitting

What's Changed
* Optionally disable rotation and translation by calebweinreb in https://github.com/dattalab/jax-moseq/pull/14
* Utf8 encoding in setup by calebweinreb in https://github.com/dattalab/jax-moseq/pull/13
* Default to float32 precision by calebweinreb in https://github.com/dattalab/jax-moseq/pull/15

**Full Changelog**: https://github.com/dattalab/jax-moseq/compare/0.0.0...0.0.1

First release

- Uses float64 precision for numerical stability
- HMM sampling in dynamax but not Kalman sampling