callbacks ========= .. Also see https://stackoverflow.com/questions/2701998/automatically-document-all-modules-recursively-with-sphinx-autodoc .. automodule:: mldft.ml.callbacks :members: :private-members: :special-members: :exclude-members: __weakref__, __hash__ .. rubric:: Modules .. autosummary:: :toctree: :template: module_template.rst :recursive: base checkpoint custom_scalars image_logging log_gradient_norm mesh_logging print_overrides_callback sub_model_summary timing