dirichlet_relative_likelihood.jax

JAX implementation for Dirichlet relative likelihood scores.

Functions

jax_compute_dirichlet_rl_score	Compute the Dirichlet relative likelihood score using JAX Arrays.
jax_compute_dirichlet_rl_score_dirichlet	Compute the score from a JaxDirichletDistribution.