/ common / utils / __init__.py
__init__.py
 1  # Utility functions
 2  from .config_loader import (
 3      get_module_config,
 4      load_benchmark_config,
 5      load_module_defaults,
 6      load_yaml,
 7  )
 8  from .model_loader import ModelInfo, export_to_onnx, get_model_info, load_model
 9  from .path_utils import ensure_project_root_in_path, get_project_root
10  from .gelu import (
11      GELU_COEF,
12      GELU_SQRT_2_OVER_PI,
13      gelu_backward,
14      gelu_erf,
15      gelu_tanh_approx,
16  )
17  
18  __all__ = [
19      # Model loading
20      "load_model",
21      "get_model_info",
22      "export_to_onnx",
23      "ModelInfo",
24      # Config loading
25      "load_yaml",
26      "load_benchmark_config",
27      "load_module_defaults",
28      "get_module_config",
29      # Path utilities
30      "get_project_root",
31      "ensure_project_root_in_path",
32      # GELU implementations
33      "gelu_tanh_approx",
34      "gelu_erf",
35      "gelu_backward",
36      "GELU_SQRT_2_OVER_PI",
37      "GELU_COEF",
38  ]