__init__.py

from .azure import Azure
from .hooks import register_azure_hooks, GradientClippingHook, LossSpikeDetectionHook

__all__ = ['Azure', 'register_azure_hooks', 'GradientClippingHook', 'LossSpikeDetectionHook']