search

lindermanlab / ssm

Bayesian learning and inference for state space models
MIT License
540 stars 196 forks source link

using GPU #146

Closed jwk0117 closed 1 year ago

jwk0117 commented 2 years ago

Hi,

Thank you for the package. I was wondering if the latest version automatically utilizes a gpu? If not, could you provide some advice on how to do so?

slinderman commented 1 year ago

Please keep an eye on an upcoming JAX library for state space modeling! https://github.com/probml/ssm-jax/