ColossalAI/colossalai/nn
CsRic 964123ae0f
[embedding] freq_aware_embedding: add small functions for caller application (#1537)
2022-09-05 15:12:53 +08:00
..
_ops [embedding] tablewise sharding polish (#1535) 2022-09-02 11:09:37 +08:00
graph [NFC] polish doc style for ColoTensor (#1457) 2022-08-16 09:21:05 +08:00
layer [NFC] polish colossalai/nn/layer/wrapper/pipeline_wrapper.py code style (#1303) 2022-07-13 19:01:07 +08:00
loss
lr_scheduler [NFC] polish colossalai/nn/lr_scheduler/onecycle.py code style (#1269) 2022-07-13 12:08:21 +08:00
metric
optimizer fix nvme docstring (#1450) 2022-08-12 18:01:02 +08:00
parallel [embedding] freq_aware_embedding: add small functions for caller application (#1537) 2022-09-05 15:12:53 +08:00
__init__.py
init.py