feat(autogram): Add kwargs support

[ValerianRey]

This adds support for modules that have some keyword arguments passed to their forward method.
Seems to work on a simple example, but needs more heavy testing.

• Add kwargs support
• Add WithModuleWithStringKwarg test
• Add WithModuleWithHybridPyTreeKwarg test
• Factorize existing architectures
• Fix some typos

[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%> (ø)
src/torchjd/autogram/_vjp.py	100.00% <100.00%> (ø)

🚀 New features to boost your workflow:

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

[PierreQuinton]

Not clear to me that we do use a non-empty kwargs in the test. But I trust you checked. LGTM

[ValerianRey]

Not clear to me that we do use a non-empty kwargs in the test. But I trust you checked. LGTM

We do it in WithModuleWithHybridKwargs. Its forward method is:

def forward(self, input: Tensor) -> Tensor:
    return self.with_string_arg(s="two", input=input)

=> The hooked module called with_string_arg receives some keyword arguments s and input. Before this PR, the hook would receive nothing, according to the documentation of register_forward_hook (link):

If with_kwargs is False or not specified, the input contains only the positional arguments given to the module. Keyword arguments won’t be passed to the hooks and only to the forward.

Now, those are correctly passed to the hook.

I'm gonna add a few more architectures to test this more heavily.