binmakeswell c95e18cdb9 [NFC] polish colossalai/kernel/cuda_native/csrc/scaled_upper_triang_masked_softmax.h code style (#1270)		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	make AutoPlacementPolicy configurable (#1191)	2022-06-30 15:18:30 +08:00
kernel	[NFC] polish colossalai/kernel/cuda_native/csrc/scaled_upper_triang_masked_softmax.h code style (#1270)	2022-07-13 12:08:21 +08:00
logging	…
nn	[NFC] polish colossalai/nn/layer/parallel_3d/_operation.py code style (#1258)	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	…
constants.py	…
core.py	…
global_variables.py	…
initialize.py	…