mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-06-01 20:05:27 +00:00
6 lines
174 B
Python
6 lines
174 B
Python
from .init import colo_uniform
|
|
from .linear import colo_linear
|
|
from .element_wise import colo_mean
|
|
from .layernorm import colo_layernorm
|
|
from .loss import colo_cross_entropy
|