search

probml / dynamax

State Space Models library in JAX
https://probml.github.io/dynamax/
MIT License
636 stars 69 forks source link

[ENH] Create `StandardHMMSparseTransitions` class #295

Closed emdupre closed 4 months ago

emdupre commented 1 year ago

References #273

Initialize a new StandardHMMSparseTransitions class. This class does not use the experimental jax.experimental.sparse library (yet) but allows to only parametrize the non-zero elements in a transition matrix.

Not yet ready to merge, though feedback is welcome at any point !

emdupre commented 1 year ago

cc @slinderman I think this is ready for review !

emdupre commented 4 months ago

Closing as stalled.