constraint_bijections

jaxns.internals.constraint_bijections

Module Contents

quick_unit(x)

Quick approximation to the sigmoid.

Parameters:

x (jax.Array) – jax.Array value in (-inf, inf) open interval

Returns:

value in (0, 1) in open interval

Return type:

jax.Array

quick_unit_inverse(y)

Inverse of quick_unit.

Parameters:

y (jax.Array) – jax.Array value in (0, 1) open interval

Returns:

value in (-inf, inf) in open interval

Return type:

jax.Array