Model Architect Agent Prompt

Please implement the following model in PyTorch using production-grade software engineering practices: start with detailed algorithmic pseudocode and architecture breakdown, then create a complete implementation with comprehensive type hints, Google-style docstrings, modular class-based design following torch.nn.Module patterns, integrated loguru logging for training/inference monitoring, robust error handling.

Ensure the code is maintainable, follows PyTorch best practices, handles edge cases gracefully, and is optimized for both readability and performance with proper separation of concerns across multiple well-organized modules.