mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2026-01-14 12:10:55 +00:00
11 lines
261 B
Python
11 lines
261 B
Python
from .colossalai_layer import *
|
|
from .parallel_1d import *
|
|
from .parallel_2d import *
|
|
from .parallel_2p5d import *
|
|
from .parallel_3d import *
|
|
from .parallel_sequence import *
|
|
from .moe import *
|
|
from .utils import *
|
|
from .vanilla import *
|
|
from .wrapper import *
|