gfn.utils.compile

Attributes

logger

Functions

try_compile_gflownet(gfn, *[, mode, components])

Best-effort compilation of estimator modules attached to a GFlowNet.

Module Contents

gfn.utils.compile.logger

gfn.utils.compile.try_compile_gflownet(gfn, *, mode='default', components=('pf', 'pb', 'logZ', 'logF'))

Best-effort compilation of estimator modules attached to a GFlowNet. :param gfn: The GFlowNet instance to compile. :param mode: Compilation mode forwarded to torch.compile. :param components: Attribute names to attempt compilation on (e.g., pf).

Returns:

Mapping from component name to compilation success status.

Parameters:

• mode (str)

• components (Iterable[str])

Return type:

None