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 [doc] Fix typo under colossalai and doc(#3618) 2023-04-26 11:38:43 +08:00
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 [NFC] polish colossalai/nn/lr_scheduler/linear.py code style (#1716) 2022-10-19 12:20:51 +08:00
metric [NFC] polish colossalai/nn/metric/_utils.py code style (#1727) 2022-10-19 12:20:51 +08:00
optimizer [nfc]fix typo colossalai/pipeline tensor nn (#3899) 2023-06-06 14:07:36 +08:00
parallel [nfc] fix typo colossalai/nn (#3887) 2023-06-05 16:04:27 +08:00
__init__.py [kernel] added jit warmup (#1792) 2022-11-08 16:22:23 +08:00
init.py [NFC] polish colossalai/nn/init.py code style (#1292) 2022-07-13 10:51:55 +08:00