You are viewing the development documentation (main). Switch to the stable release docs.

wasserstein_distance.jax¶

JAX implementation for Wasserstein distance scores.

Functions¶

Computes the Wasserstein distance score using JAX arrays.