modules moved to equinox

[astanziola]

Fixes #121

ToDo before merging:

• [x] Test against jwave
• [x] Test all notebooks
• [x] Add an example of the use of .θ notation for parameters
• [x] Add types
• [x] Check that the documentation correctly compiles

[codecov[bot]]

Codecov Report

Attention: 69 lines in your changes are missing coverage. Please review.

Comparison is base (ee5ba44) 91.41% compared to head (c124817) 92.88%.

Files	Patch %	Lines
jaxdf/operators/differential.py	75.36%	34 Missing :warning:
jaxdf/operators/functions.py	71.62%	21 Missing :warning:
jaxdf/discretization.py	96.00%	5 Missing :warning:
jaxdf/operators/magic.py	95.34%	4 Missing :warning:
jaxdf/geometry.py	96.61%	2 Missing :warning:
jaxdf/util.py	87.50%	2 Missing :warning:
jaxdf/core.py	99.20%	1 Missing :warning:

Additional details and impacted files

```diff @@ Coverage Diff @@ ## main #139 +/- ## ========================================== + Coverage 91.41% 92.88% +1.47% ========================================== Files 13 15 +2 Lines 943 970 +27 ========================================== + Hits 862 901 +39 + Misses 81 69 -12 ```

:umbrella: View full report in Codecov by Sentry.
:loudspeaker: Have feedback on the report? Share it here.