A unified framework for marginal and conditional two-sample testing with kernels in JAX.
Reproducing Kernel Hilbert Space: kernels, MMD, and kernel quantile embeddings (re-exports hopfield for AM)