dev: support numpy>=2.0.0

[galah92]

Consider the following:

uv add gpjax --optional jax
  × No solution found when resolving dependencies:
  ╰─▶ Because gpjax==0.9.1 depends on numpy<2.0.0 and only gpjax<=0.9.1 is available, we can conclude that gpjax>=0.9.1 depends on numpy<2.0.0.
      And because pyproj[jax] depends on gpjax>=0.9.1, we can conclude that pyproj[jax] depends on numpy<2.0.0.
      And because your project depends on numpy>=2.1.1 and your project requires pyproj[jax], we can conclude that your projects's requirements are unsatisfiable.
  help: If this is intentional, run `uv add --frozen` to skip the lock and sync steps.

Can gpjax support numpy>=2.0.0?

[thomaspinder]

Hey @galah92 , we're unable to move to numpy2.0 due to TensorFlow-Probability. Relevant issue: tensorflow/probability#1814

[galah92]

Understood, thank you.

[github-actions]

There has been no recent activity on this issue. To keep our issues log clean, we remove old and inactive issues.
Please update to the latest version of GPJax and check if that resolves the issue. Let us know if that works for you by leaving a comment.
This issue is now marked as stale and will be closed if no further activity occurs. If you believe that this is incorrect, please comment. Thank you!