mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2026-05-03 17:37:23 +00:00
4 lines
136 B
Python
4 lines
136 B
Python
from .init import colo_uniform
|
|
from .linear import colo_linear
|
|
from .element_wise import colo_mean
|
|
from .layernorm import colo_layernorm |