dattalab / keypoint-moseq

https://keypoint-moseq.readthedocs.io
Other
64 stars 26 forks source link

Added parallel_message_passing flag #77

Closed kaijfox closed 11 months ago

kaijfox commented 11 months ago

Added parallel_message_passing flag in fit_model to use the new parallel posterior sampler in jax-moseq (dattalab/jax-moseq#23) and dynamax (probml/dynamax#324) when in a suitable parallel computing environment.