ColossalAI/colossalai/nn
binmakeswell 089c365fa0
[doc] add Series A Funding and NeurIPS news (#4377)
* [doc] add Series A Funding and NeurIPS news

* [kernal] fix mha kernal

* [CI] skip moe

* [CI] fix requirements
2023-08-04 17:42:07 +08:00
..
_ops
layer [shardformer] integrated linear 1D with dtensor (#3996) 2023-07-04 16:05:01 +08:00
loss [nfc] fix typo colossalai/nn (#3887) 2023-06-05 16:04:27 +08:00
lr_scheduler
metric
optimizer [doc] add Series A Funding and NeurIPS news (#4377) 2023-08-04 17:42:07 +08:00
parallel [nfc] fix typo colossalai/nn (#3887) 2023-06-05 16:04:27 +08:00
__init__.py
init.py