kullback_leibler.jax

JAX implementation for Kullback-Leibler divergence scores.

Functions

jax_compute_kl_divergence_score

Computes the Kullback-Leibler divergence score using JAX Array.