mirror of https://github.com/hpcaitech/ColossalAI
f2d2a1597a | ||
---|---|---|
.. | ||
csrc | ||
__init__.py | ||
layer_norm.py | ||
multihead_attention.py | ||
scaled_softmax.py |
f2d2a1597a | ||
---|---|---|
.. | ||
csrc | ||
__init__.py | ||
layer_norm.py | ||
multihead_attention.py | ||
scaled_softmax.py |