JAX support for refactor

[obackhouse]

Tentative support for JAX as backend

[codecov]

Codecov Report

❌ Patch coverage is 75.00000% with 32 lines in your changes missing coverage. Please review.
✅ Project coverage is 72.45%. Comparing base (acc43ff) to head (3b6c18a).

Files with missing lines	Patch %	Lines
dyson/_backend.py	83.33%	7 Missing and 2 partials ⚠️
dyson/expressions/adc.py	27.27%	8 Missing ⚠️
dyson/solvers/static/davidson.py	44.44%	4 Missing and 1 partial ⚠️
dyson/solvers/static/density.py	37.50%	4 Missing and 1 partial ⚠️
dyson/solvers/dynamic/corrvec.py	25.00%	2 Missing and 1 partial ⚠️
dyson/expressions/hamiltonian.py	71.42%	2 Missing ⚠️

Additional details and impacted files

@@             Coverage Diff              @@
##           refactor      #13      +/-   ##
============================================
- Coverage     72.49%   72.45%   -0.05%     
============================================
  Files            38       39       +1     
  Lines          3861     3964     +103     
  Branches        488      497       +9     
============================================
+ Hits           2799     2872      +73     
- Misses          863      888      +25     
- Partials        199      204       +5     

☔ 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.