def gauss_at
import mlbend-generative@0.1.1.0/main.bend as MlbendGenerativeimport Base
80def gauss_at(+s: Nat, +j: Nat) -> F32:
81 +pair = Nat.sub(j, Nat.mod(j, 2n))
82 +ua = uniform(s, pair)
83 +ub = uniform(s, (1n + pair : Nat))
84 +r = F32.sqrt((0.0 - (2.0 * F32.log(ua) : F32) : F32))
85 gauss_body(Nat.is_eq(Nat.mod(j, 2n), 0n), r, (6.2831855 * ub : F32))