ColossalAI/colossalai/nn
Jiarui Fang 4b9bba8116
[ColoTensor] rename APIs and add output_replicate to ComputeSpec (#1168)
2022-06-24 13:08:54 +08:00
..
_ops [ColoTensor] rename APIs and add output_replicate to ComputeSpec (#1168) 2022-06-24 13:08:54 +08:00
graph
layer
loss
lr_scheduler
metric
optimizer
parallel [ColoTensor] rename APIs and add output_replicate to ComputeSpec (#1168) 2022-06-24 13:08:54 +08:00
__init__.py
init.py