"""
Training module for DP-SGD Explorer.

Contains:
- MockTrainer: Simulation-based training for fast experimentation
- SimplifiedRealTrainer: Real TensorFlow-based DP-SGD training
- RealTrainer: Full TensorFlow Privacy-based DP-SGD training
- PrivacyCalculator: Unified RDP-based privacy accounting
- gradient_utils: Shared gradient visualization utilities
"""

from .mock_trainer import MockTrainer
from .privacy_calculator import PrivacyCalculator, get_privacy_calculator
from .gradient_utils import (
    generate_gradient_norms,
    generate_clipped_gradients,
    generate_gradient_info
)

__all__ = [
    'MockTrainer',
    'PrivacyCalculator',
    'get_privacy_calculator',
    'generate_gradient_norms',
    'generate_clipped_gradients',
    'generate_gradient_info',
]