import moe._custom_ops as ops