jcm/sde_lib.py [193:202]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        return drift, diffusion

    def marginal_prob(self, x, t, high_precision=True):
        log_mean_coeff = (
            -0.25 * t**2 * (self.beta_1 - self.beta_0) - 0.5 * t * self.beta_0
        )
        if high_precision:
            mean = batch_mul(
                jnp.where(
                    jnp.abs(log_mean_coeff) <= 1e-3,
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



jcm/sde_lib.py [306:315]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        return drift, diffusion

    def marginal_prob(self, x, t, high_precision=True):
        log_mean_coeff = (
            -0.25 * t**2 * (self.beta_1 - self.beta_0) - 0.5 * t * self.beta_0
        )
        if high_precision:
            mean = batch_mul(
                jnp.where(
                    jnp.abs(log_mean_coeff) <= 1e-3,
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -