Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
__init__.py 97 Bytes
Newer Older
Rick Ho's avatar
reconstruct fmoe nn module  
Rick Ho committed
1 
from .moe import BruteForceMoE
Rick Ho's avatar
fix python bugs  
Rick Ho committed
2 
from .layers import FMoELinear, FMoENaiveGate, FMoETransformerMLP