Generalized loss terms from trial specifications about feedbax HOT 2 CLOSED

A working implemention is available as of 4f735f4.

In feedbax.loss, [TargetStateLoss][https://github.com/mlprt/feedbax/blob/4f735f434ededd122ddeee6957fd911a9e8870c7/feedbax/loss.py#L357] is a subclass of AbstractLoss that associates a where, a norm function, and a TargetSpec.

A TargetSpec provides information about 1) the target value of the state, 2) the time indices at which the state's value should be penalized (e.g. penalize effector velocity on final time step only), and 3) an array of discounting factors. All of these fields are optional, and partial specifications may be combined.

In TaskTrialSpec, there is now a field targets: WhereDict[TargetSpec] through which a subclass of AbstractLoss can provide trial-by-trial TargetSpec information to instances of TargetStateLoss (via TaskTrainer).

When a TargetStateLoss instance is called, its spec: Optional[TargetSpec] field is eqx.combine'd with any entries in trial_specs.targets. This allows the user to supply default target values on instantiation of TargetStateLoss, but also for the task to be designed (example) so that it provides trial-by-trial targets. A target value must be specified either trial-by-trial or as a default -- an error is raised if no target value is available.

A loss_func must still be passed on instantiating an AbstractTask subclass. Composing the terms of the loss function is now a little more complicated than it used to be, since we add TargetStateLoss instances by specifying the part of the state to penalize. We can probably replace the old loss classes like EffectorPositionLoss with factories/wrappers for TargetStateLoss, which would simplify the loss function construction again, in some typical use cases.

from feedbax.

One issue that remains is the possibility of multiple targets being specified with respect to the same part of the state. For example, in delayed reaching we might want a separate loss terms for the effector position error with respect to 1) the reach goal, and 2) the initial fixation. The possibility of multiple loss terms on a single target is why I enabled tuple[Callable, str] keys for WhereDict, so that a where lambda can be combined with a unique label. However, this means that we have to make sure that the label field of TargetStateLoss matches the string entry in TargetSpecs constructed by the task. There are a couple of other options here:

1. Allow the values of trial_specs.targets to be a Mapping[str, TargetSpec]. This doesn't solve the string-matching issue, but it does simplify the allowable keys of WhereDict.
2. Only allow a single target for each part of the state. This should be possible (e.g. goal and fixation targets happen at different times during the reach) but it would mean that some other mechanism (say, in AbstractLoss) would be necessary if we want users to be able to distinguish loss contributions from different epochs of a trial.

from feedbax.