Similar repositories to josauder/jax-svgd: