search

probml / dynamax

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

Allow Numpy 2.0 #366

Open slinderman opened 3 months ago

slinderman commented 3 months ago

Tensorflow probability broke for numpy 2.0. See https://github.com/tensorflow/probability/issues/1814. Once that issue is resolved, we can unpin the numpy<2.0 requirement in setup.cfg.