Warn that in future arguments after the initial (prediction, ground_truth) positional arguments will become keyword-only in optax losses.
This is to prevent users from accidentally passing in the wrong order of arguments, and to make interfaces functionally more similar (e.g. after partialling loss-specific hyper-parameters).
Warn that in future arguments after the initial (prediction, ground_truth) positional arguments will become keyword-only in optax losses.
This is to prevent users from accidentally passing in the wrong order of arguments, and to make interfaces functionally more similar (e.g. after partialling loss-specific hyper-parameters).