feat: export JSON schema

[b8raoult]

Description

Add the command anemoi-training schema to export the JSON schema of a training config. Useful for validation tools or IDE such as vscode, that will use it to autocomplete and highlight errors.

What problem does this change solve?

What issue or task does this change relate to?

Additional notes

As a contributor to the Anemoi framework, please ensure that your changes include unit tests, updates to any affected dependencies and documentation, and have been tested in a parallel setting (i.e., with multiple GPUs). As a reviewer, you are also responsible for verifying these aspects and requesting changes if they are not adequately addressed. For guidelines about those please refer to https://anemoi.readthedocs.io/en/latest/

By opening this pull request, I affirm that all authors agree to the Contributor License Agreement.

[frazane]

Perhaps better as a subcommand of anemoi-training config?

[anaprietonem]

Perhaps better as a subcommand of anemoi-training config?

Hey @frazane good suggestion, agreed with it. I had a go, could you review it and see if that's what you have in mind?

[frazane]

Hi @anaprietonem, I think this makes sense!

On a more general note: how do we plan to use the schema? I am assuming it cannot be used in combination with structured configs, but only with "unrolled" ones. Is that so? Perhaps a few lines explaining how these schemas are intended to be used (on a high level) would help give them more visibility so more people will experiment with them.