fix: Make ML __init__.py use lazy imports to avoid import errors

feat: Add ML module __init__.py