Forward solver adapted for automatic differentiation.

[Ig-dolci]

This PR seeks to update the acoustic forward and FWI code using the automated adjoint solver.

New tools:

• Two-level parallelism: Mesh and source with firedrake.adjoint.EnsembleReducedFunctional
• Enable checkpointing.

What is missing: a proper absorbing boundary condition.

[codecov]

Codecov Report

Attention: Patch coverage is 91.02564% with 7 lines in your changes missing coverage. Please review.

Project coverage is 83.86%. Comparing base (2f19564) to head (3ed4881).
Report is 14 commits behind head on main.

Files with missing lines	Patch %	Lines
spyro/solvers/forward_ad.py	87.71%	7 Missing ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##             main     #124      +/-   ##
==========================================
+ Coverage   83.69%   83.86%   +0.16%     
==========================================
  Files          47       49       +2     
  Lines        3361     3439      +78     
==========================================
+ Hits         2813     2884      +71     
- Misses        548      555       +7     

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

[Ig-dolci]

Spyro needs to update the firedrake version to have the checkpointing test pass.

[Olender]

Spyro needs to update the firedrake version to have the checkpointing test pass.

I can install the latest version on the runner. We're currently using the version from 2024-07-19. Please let me know which version we should target for this update.

[Ig-dolci]

Spyro needs to update the firedrake version to have the checkpointing test pass.

I can install the latest version on the runner. We're currently using the version from 2024-07-19. Please let me know which version we should target for this update.

The latest.

[Olender]

Ok, I have already installed everything in the runner and changed the firedrake location in the .yml workflow file