Remove old JAX restrictions for Winows and test cleanup

[kratman]

Description

Support for JAX on windows was improved. This change removes the nox file restrictions on JAX. This builds on #3932 and keeps that one focused on just ODES.

Fixes #3941

Key checklist:

• No style issues: $ pre-commit run (or $ nox -s pre-commit) (see CONTRIBUTING.md for how to set this up to run automatically when committing locally, in just two lines of code)
• All tests pass: $ python run-tests.py --all (or $ nox -s tests)
• The documentation builds: $ python run-tests.py --doctest (or $ nox -s doctests)

You can run integration tests, unit tests, and doctests together at once, using $ python run-tests.py --quick (or $ nox -s quick).

Further checks:

• Code is commented, particularly in hard-to-understand areas
• Tests added that prove fix is effective or that feature works

[kratman]

This will be rebased/forced pushed after #3932 is merged

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 99.58%. Comparing base (a3db966) to head (c6467f1).
Report is 2 commits behind head on develop.

❗ Current head c6467f1 differs from pull request most recent head 4bf5a3a. Consider uploading reports for the commit 4bf5a3a to get more accurate results

Additional details and impacted files

@@             Coverage Diff             @@
##           develop    #3942      +/-   ##
===========================================
- Coverage    99.60%   99.58%   -0.03%     
===========================================
  Files          259      257       -2     
  Lines        21347    21190     -157     
===========================================
- Hits         21263    21102     -161     
- Misses          84       88       +4     

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

[kratman]

I broke this branch when I rebased. Closing