jax_compute_tv_score

probly.conformal_scores.total_variation.jax.jax_compute_tv_score(y_pred: Array, y_true: Array) Array[source]

Computes the Total Variation score using JAX arrays.