mirror of https://github.com/hpcaitech/ColossalAI
aibig-modeldata-parallelismdeep-learningdistributed-computingfoundation-modelsheterogeneous-traininghpcinferencelarge-scalemodel-parallelismpipeline-parallelism
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ver217
f8a7148dec
|
2 years ago | |
---|---|---|
.. | ||
__init__.pyi | 2 years ago | |
cpu_optim.pyi | 2 years ago | |
fused_optim.pyi | 2 years ago | |
layer_norm.pyi | 2 years ago | |
moe.pyi | 2 years ago | |
multihead_attention.pyi | 2 years ago | |
scaled_masked_softmax.pyi | 2 years ago | |
scaled_upper_triang_masked_softmax.pyi | 2 years ago |