scoring_rule.jax

JAX implementations of scoring rule loss vectors.

Functions

jax_brier_loss_vector	Compute the per-label Brier loss vector for a JAX array.
jax_log_loss_vector	Compute the per-label log loss vector for a JAX array.
jax_spherical_loss_vector	Compute the per-label spherical loss vector for a JAX array.
jax_zero_one_loss_vector	Compute the per-label zero-one loss vector for a JAX array.