Is your feature request related to a problem? Please describe.

Allow optional training with optimisation towards tendency as opposed to state

Describe the solution you'd like

Adapt logic to allow the objective to be training towards tendencies

Describe alternatives you've considered

NA

Additional context

Has been shown to improve results with a diffusion model targeting atmosphere - this will be useful for investigating whether or not improvements extend to other models and other domains such as ocean and wave.

Besides implement logic key things to consider:

1. Any adjustments required to Imputer logic?
2. [Please Suggest More]

Organisation

ECMWF