feat: Numpyro integration

[thomaspinder]

Let's seek to migrate our TFP functionality to Numpyro. From here, we can seek to integrate GPJax within numpyro.

[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!

[github-actions]

There has been no activity on this PR for some time. Therefore, we will be automatically closing the PR if no new activity occurs within the next seven days.
Thank you for your contributions.