mirror of https://github.com/hpcaitech/ColossalAI
f8a7148dec | ||
---|---|---|
.. | ||
__init__.pyi | ||
cpu_optim.pyi | ||
fused_optim.pyi | ||
layer_norm.pyi | ||
moe.pyi | ||
multihead_attention.pyi | ||
scaled_masked_softmax.pyi | ||
scaled_upper_triang_masked_softmax.pyi |
f8a7148dec | ||
---|---|---|
.. | ||
__init__.pyi | ||
cpu_optim.pyi | ||
fused_optim.pyi | ||
layer_norm.pyi | ||
moe.pyi | ||
multihead_attention.pyi | ||
scaled_masked_softmax.pyi | ||
scaled_upper_triang_masked_softmax.pyi |