bilby.compat.jaxΒΆ Classes JittedLikelihood(likelihood[, cast_to_float]) A wrapper to just-in-time compile a Bilby likelihood for use with jax.