:modulename:`compose.utils`
============================================

.. currentmodule:: chemtrain.compose.utils

.. autoclass:: ApplyFn
   :members:

.. autofunction:: batch_apply_fn

.. autofunction:: flatten_graph