Upgrade to jax>=0.6.2

[arturtoshev]

• Python from ">=3.9" to ">=3.10"
• JAX from "==0.4.29" to ">=0.6.2"
• Fix loading r0 properties

[codecov-commenter]

Codecov Report

❌ Patch coverage is 0% with 3 lines in your changes missing coverage. Please review.
✅ Project coverage is 68.0%. Comparing base (4c6f978) to head (090bdea).

Files with missing lines	Patch %	Lines
jax_sph/defaults.py	0.0%	2 Missing ⚠️
jax_sph/case_setup.py	0.0%	1 Missing ⚠️

❌ Your patch status has failed because the patch coverage (0.0%) is below the target coverage (50.0%). You can increase the patch coverage or adjust the target coverage.

Additional details and impacted files

@@          Coverage Diff          @@
##            main     #21   +/-   ##
=====================================
  Coverage   68.0%   68.0%           
=====================================
  Files         11      11           
  Lines       1420    1420           
=====================================
  Hits         967     967           
  Misses       453     453           

Flag	Coverage Δ
unittests	68.0% <0.0%> (ø)

Flags with carried forward coverage won't be shown. Click here to find out more.

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:

• ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

[arturtoshev]

Using this JAX-SPH version in LagrangeBench might require some adjustments. But this is on the lagrangebench side.