mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2026-07-12 10:54:49 +00:00
5 lines
172 B
Python
5 lines
172 B
Python
from .linear import colo_linear
|
|
from .element_wise import *
|
|
from .layernorm import colo_layernorm
|
|
from .loss import colo_cross_entropy
|
|
from .embedding import colo_embedding |