dynamax
dynamax copied to clipboard
Allow Numpy 2.0
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.
This is still broken.
At this point I think it's fair to assume that numpy < 2.0 will be a requirement of dynamax for the foreseeable future. It doesn't seem to be a high priority item for the tensorflow probability team.