Added capabilities to run an AR-HMM with robust autoregression.
I haven't tested these functions to make sure they produce the exact same results as pybasicbayes, but I closely followed the algorithms written in pybasicbayes, and reformatted some places to allow the jax-moseq version to work on all states simultaneously.
I'm sure there are ways we can optimize the code written here, as this was a first attempt.
Added capabilities to run an AR-HMM with robust autoregression.
I haven't tested these functions to make sure they produce the exact same results as pybasicbayes, but I closely followed the algorithms written in pybasicbayes, and reformatted some places to allow the
jax-moseq
version to work on all states simultaneously.I'm sure there are ways we can optimize the code written here, as this was a first attempt.