gfn.utils.compile
=================

.. py:module:: gfn.utils.compile


Attributes
----------

.. autoapisummary::

   gfn.utils.compile.logger


Functions
---------

.. autoapisummary::

   gfn.utils.compile.try_compile_gflownet


Module Contents
---------------

.. py:data:: logger

.. py:function:: 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.