ColossalAI/colossalai/nn
Frank Lee 015af592f8 [shardformer] integrated linear 1D with dtensor (#3996)
* [shardformer] integrated linear 1D with dtensor

* polish code
2023-07-04 16:05:01 +08:00
..
_ops
layer [shardformer] integrated linear 1D with dtensor (#3996) 2023-07-04 16:05:01 +08:00
loss
lr_scheduler
metric
optimizer [nfc]fix typo colossalai/pipeline tensor nn (#3899) 2023-06-06 14:07:36 +08:00
parallel
__init__.py
init.py