mirror of https://github.com/hpcaitech/ColossalAI
7806842f2d | ||
---|---|---|
.. | ||
cpu_adam | ||
flash_attention | ||
inference | ||
layernorm | ||
moe | ||
optimizer | ||
softmax | ||
__init__.py |
7806842f2d | ||
---|---|---|
.. | ||
cpu_adam | ||
flash_attention | ||
inference | ||
layernorm | ||
moe | ||
optimizer | ||
softmax | ||
__init__.py |