xomponents Component discovery models (decomposition, mixture) in JAX with a scikit-learn-like interface