refactor(autogram): Switch to ComputeModuleJacobians

[PierreQuinton]

• Make AccumulateJacobian return the Jacobian w.r.t. the parameters of the module rather than providing them to the GramianAccumulator directly.
• Rename the class AccumulateJacobian to ComputeModuleJacobians to better reflect its role.

[codecov]

Codecov Report

✅ All modified and coverable lines are covered by tests.

Files with missing lines	Coverage Δ
src/torchjd/autogram/_module_hook_manager.py	100.00% <100.00%> (ø)

🚀 New features to boost your workflow:

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

[ValerianRey]

todos are outdated right?

[ValerianRey]

I think this PR should be grouped with the other PRs that you have in mind, so that we can get impactful changes together with this.

[PierreQuinton]

todos are outdated right?

No, I just put them in the wrong PR.

I think this PR should be grouped with the other PRs that you have in mind, so that we can get impactful changes together with this.

Possibly, but this is a strict improvement that is pretty atomic and independent of anything I have in mind so far (but typically a requirement), it makes it easy to review. Whatever you prefer.

[ValerianRey]

todos are outdated right?

No, I just put them in the wrong PR.

I think this PR should be grouped with the other PRs that you have in mind, so that we can get impactful changes together with this.

Possibly, but this is a strict improvement that is pretty atomic and independent of anything I have in mind so far (but typically a requirement), it makes it easy to review. Whatever you prefer.

Yes I understand better now and I agree.