Set `parallel_message_passing` flag in jax_moseq

dattalab / keypoint-moseq

https://keypoint-moseq.readthedocs.io

Other

64 stars 26 forks source link

Set `parallel_message_passing` flag in jax_moseq #65

Closed calebweinreb closed 11 months ago

calebweinreb commented 1 year ago

When https://github.com/dattalab/jax-moseq/pull/23 goes through, we should bump the pinned jax-moseq version in setup.cfg and also add a parallel_message_passing flag. If the flag is None, serial vs. parallel should be determined automatically depending whether the caller has a access to a GPU.

calebweinreb commented 11 months ago

Done by Kai: https://github.com/dattalab/keypoint-moseq/pull/77