Implement Arcee Trinity Coarse-Grained MoE FFN in Python