ColossalAI/colossalai/nn
digger yu 9265f2d4d7
[NFC]fix typo colossalai/auto_parallel nn utils etc. (#3779)
* fix typo colossalai/autochunk auto_parallel amp

* fix typo colossalai/auto_parallel nn utils etc.
2023-05-23 15:28:20 +08:00
..
_ops [doc] Fix typo under colossalai and doc(#3618) 2023-04-26 11:38:43 +08:00
layer [doc] Fix typo under colossalai and doc(#3618) 2023-04-26 11:38:43 +08:00
loss [NFC] polish colossalai/nn/loss/loss_2p5d.py code style (#1553) 2022-09-08 22:11:04 +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/auto_parallel nn utils etc. (#3779) 2023-05-23 15:28:20 +08:00
parallel [NFC]fix typo colossalai/auto_parallel nn utils etc. (#3779) 2023-05-23 15:28:20 +08:00
__init__.py [kernel] added jit warmup (#1792) 2022-11-08 16:22:23 +08:00
init.py