ColossalAI/colossalai/nn
アマデウス 4268ae017b
[kernel] added jit warmup (#1792)
2022-11-08 16:22:23 +08:00
..
_ops [hotfix[ fix colotensor.type() raise NotImplementedError (#1682) 2022-10-10 10:13:31 +08:00
graph
layer [kernel] added jit warmup (#1792) 2022-11-08 16:22:23 +08:00
loss
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 add optimizer README for tutorials (#1707) 2022-10-14 09:10:18 +00:00
parallel [Gemini] make gemini usage simple (#1821) 2022-11-08 15:53:13 +08:00
__init__.py [kernel] added jit warmup (#1792) 2022-11-08 16:22:23 +08:00
init.py