zbian 598d456d0e fixed logger		2022-11-15 16:00:07 +08:00
..
amp	[NFC] polish colossalai/amp/naive_amp/__init__.py code style (#1905)	2022-11-11 17:49:18 +08:00
auto_parallel	[autoparallel] remove redundancy comm node (#1893)	2022-11-15 10:53:41 +08:00
builder	[NFC] polish colossalai/builder/__init__.py code style (#1560)	2022-09-08 22:11:04 +08:00
cli	…
communication	[communication] add p2p_v2.py to support communication with List[Any] (#1407)	2022-08-09 11:40:04 +08:00
context	updated tp layers	2022-11-02 12:19:38 +08:00
device	[autoparallel] add numerical test for node strategies (#1760)	2022-10-27 10:42:54 +08:00
engine	[engin/schedule] use p2p_v2 to recontruct pipeline_schedule (#1408)	2022-08-12 11:33:26 +08:00
fx	[SC] add GPT example for auto checkpoint (#1889)	2022-11-11 23:17:25 +08:00
gemini	MemStatsCollectorStatic (#1765)	2022-11-07 16:49:03 +08:00
kernel	updated flash attention api	2022-11-15 15:25:39 +08:00
logging	fixed logger	2022-11-15 16:00:07 +08:00
nn	[tensorparallel] fixed tp layers (#1938)	2022-11-14 17:34:03 +08:00
pipeline	[Pipeline]Adapt to Pipelinable OPT (#1782)	2022-11-01 14:18:50 +08:00
registry	…
tensor	[autoparallel] remove redundancy comm node (#1893)	2022-11-15 10:53:41 +08:00
testing	[unittest] added doc for the pytest wrapper (#1704)	2022-10-14 10:56:17 +08:00
trainer	[NFC] polish _checkpoint_hook.py code style (#1722)	2022-10-19 12:20:51 +08:00
utils	[ColoTensor] ColoInitContext initialize parameters in shard mode. (#1937)	2022-11-14 16:05:09 +08:00
zero	[zero] migrate zero1&2 (#1878)	2022-11-11 09:26:40 +08:00
__init__.py	version to 0.1.11rc2 (#1832)	2022-11-08 17:25:15 +08:00
constants.py	updated tp layers	2022-11-02 12:19:38 +08:00
core.py	…
global_variables.py	updated tp layers	2022-11-02 12:19:38 +08:00
initialize.py	…