nami.interpolants.gamma

Classes

BrownianGamma([eps])
GammaSchedule()
ScaledBrownianGamma([scale, eps])	Brownian gamma schedule with variance scale.
ZeroGamma()

class nami.interpolants.gamma.BrownianGamma(eps=1e-12)

Bases: GammaSchedule

Parameters:

eps (float)

eps: float = 1e-12

gamma(t)

Return type:

Tensor

Parameters:

t (Tensor)

gamma_dot(t)

Return type:

Tensor

Parameters:

t (Tensor)

gamma_gamma_dot(t)

Return type:

Tensor

Parameters:

t (Tensor)

class nami.interpolants.gamma.GammaSchedule

Bases: object

gamma(t)

Return type:

Tensor

Parameters:

t (Tensor)

gamma_dot(t)

Return type:

Tensor

Parameters:

t (Tensor)

gamma_gamma_dot(t)

Return type:

Tensor

Parameters:

t (Tensor)

class nami.interpolants.gamma.ScaledBrownianGamma(scale=1.0, eps=1e-12)

Bases: GammaSchedule

Brownian gamma schedule with variance scale.

The schedule is gamma(t)^2 = scale * t * (1 - t). For bridge paths parameterised by sigma where gamma(t) = sigma * sqrt(t * (1 - t)), use scale = sigma**2 (or from_sigma).

Parameters:

• scale (float)

• eps (float)

eps: float = 1e-12

classmethod from_sigma(sigma, eps=1e-12)

Construct a schedule with gamma(t) = sigma * sqrt(t * (1 - t)).

Return type:

ScaledBrownianGamma

Parameters:

• sigma (float)

• eps (float)

gamma(t)

Return type:

Tensor

Parameters:

t (Tensor)

gamma_dot(t)

Return type:

Tensor

Parameters:

t (Tensor)

gamma_gamma_dot(t)

Return type:

Tensor

Parameters:

t (Tensor)

scale: float = 1.0

class nami.interpolants.gamma.ZeroGamma

Bases: GammaSchedule

gamma(t)

Return type:

Tensor

Parameters:

t (Tensor)

gamma_dot(t)

Return type:

Tensor

Parameters:

t (Tensor)

gamma_gamma_dot(t)

Return type:

Tensor

Parameters:

t (Tensor)