flowmc (pypi): Normalizing flow exhanced sampler in jax
Keywords: sampling|inference|machine learning|normalizing|autodiff|jax|mcmc|normalizing-flows|stat