Maruyama_Aya 87f679aeae [NFC] polish colossalai/kernel/cuda_native/csrc/kernels/include/kernels.h code style (#1291)		2022-07-13 12:08:21 +08:00
..
amp	…
builder	[NFC] polish colossalai/builder/builder.py code style (#1265)	2022-07-13 12:08:21 +08:00
cli	…
communication	[NFC] polish colossalai/communication/collective.py (#1262)	2022-07-13 12:08:21 +08:00
context	…
engine	[NFC] polish colossalai/engine/ophooks/utils.py code style (#1256)	2022-07-13 12:08:21 +08:00
fx	[fx] added ndim property to proxy (#1253)	2022-07-12 15:27:13 +08:00
gemini	…
kernel	[NFC] polish colossalai/kernel/cuda_native/csrc/kernels/include/kernels.h code style (#1291)	2022-07-13 12:08:21 +08:00
logging	…
nn	[NFC] polish colossalai/nn/layer/parallel_1d/layers.py code style (#1290)	2022-07-13 12:08:21 +08:00
pipeline	…
registry	…
tensor	[hotfix] Dist Mgr gather torch version (#1284)	2022-07-13 00:18:56 +08:00
testing	…
trainer	…
utils	[tensor] distributed checkpointing for parameters (#1240)	2022-07-12 15:51:06 +08:00
zero	[hotfix] fix sharded optim step and clip_grad_norm (#1226)	2022-07-08 13:34:48 +08:00
__init__.py	[NFC] polish colossalai/__init__.py code style (#1285)	2022-07-13 12:08:21 +08:00
constants.py	…
core.py	…
global_variables.py	…
initialize.py	…