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.