optimum/habana/transformers/models/gpt_neox/__init__.py

from .modeling_gpt_neox import (
    GaudiGPTNeoXAttention,
    GaudiGPTNeoXForCausalLM,
    GaudiGPTNeoXLayer,
    gaudi_gpt_neox_model_forward,
)