Losses

build_loss_functions

build_loss_functions(spec)

Return per-item and per-element loss functions honoring spec overrides.

Parameters:

Name	Type	Description	Default
spec	object	Model specification exposing dynamics_module and eval_loss_kwargs.	required

Returns:

Type	Description
tuple	(per_item_fn, per_element_fn) where each callable maps (predictions, targets) to loss arrays.