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

from .modeling_esmfold import (
    gaudi_esm_for_protein_folding_forward,
    gaudi_esmfolding_trunk_forward,
    gaudi_rot_matmul,
    gaudi_rot_vec_mul,
)