JAX MoSeq 0.3.0

Bug fixes and refactored installation for compatibility with CUDA12 and unpinned jax version. Use one of the following

pip install jax-moseq[cuda11]
pip install jax-moseq[cuda12]
pip install jax-moseq # cpu

What's Changed

• bugfix for nlags=1 by @calebweinreb in #36

Full Changelog: 0.2.2...0.3.0